pytorch的安装

为了方便学习深度学习，我在电脑上尝试安装了pytorch1.2版本，在安装时也出现了很多错误，希望这期分享能帮助到深度学习的初学者们。　　

环境内容

torch：1.2.0
torchvision：0.4.0

 

下载安装

Anaconda，官网地址：https://www.anaconda.com/distribution/

cuda10.0，官网的地址是：
cuda10.0官网地址
cudnn7.4.1.5。，官网的地址是：
cudnn官网地址

安装完后在C盘这个位置可以找到根目录。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
然后大家把Cudnn的内容进行解压。

解压放入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

 

配置环境变量

Win+R启动cmd，在命令提示符内输入以下命令创建环境：

conda create –n pytorch python=3.6

 

激活环境:

activate pytorch

打开pytorch的官方安装方法：
https://pytorch.org/get-started/previous-versions/
官网推荐的安装代码如下，我使用的是Cuda10的版本：

# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

 

测试