为了方便学习深度学习,我在电脑上尝试安装了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 FilesNVIDIA GPU Computing ToolkitCUDAv10.0
然后大家把Cudnn的内容进行解压。
解压放入C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.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