PyTorch安装

从0开始在win10上安装PyTorch。

PyTorch是一个类似Numpy的科学计算工具库，最大的特点是可以利用GPU进行并行计算。

1.安装Anaconda 

 这里选Anaconda3 5.3.1  其集成的是python 3.7环境

 官网很慢，从清华镜像源下载，下载链接如下：

  https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Windows-x86_64.exe

 

2.安装CUDA

注意和python、pytorch版本对应，这里选择10.2  

下载地址：

https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

 

3.安装pytorch

  这里使用pip方式，conda方式安装速度过慢，镜像源好像都不能用了。 也要注意版本对应，在pytorch官网选择相应环境，自然会给出对应的安装语句。这里用的是豆瓣镜像源。

   cmd管理员权限运行，然后输入下列安装语句，大概一个G大小。

 

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

 

4.安装pycharm社区版

https://www.jetbrains.com/pycharm/download/#section=windows

 

5.测试是否成功

 pycharm新建工程，选择Anaconda下的python作为编译器。

import torch

print(torch.__version__)  # 检测pytorch版本
print("gpu:", torch.cuda.is_available())  # 检测gpu是否可用