网上很多教程都是直接用conda安装pytorch,但是由于pytorch很大,所以下载的时候会中断,conda下载的时候一旦中断之前都白下了,所以这里介绍第二种方式。
我使用的环境是 python3.7+pytorch1.0.0+cuda10.2+cudnn7.x(后面这个x应该问题不大,我使用最新的版本)
步骤一:CUDA安装
如果是用GPU版的pytorch,首先应该安装cuda 。 cuda下载地址 。
步骤二:CUDNN安装
安装完了CUDA之后还需要安装CUDNN,cudnn下载地址,下载cudnn的时候需要登录账号,这个可以用微信登录,需要验证一下邮箱。
步骤三:下载pytorch和torchvision
最好用浏览器下载,下载pytorch的时候会中断几次,需要手动点击继续下载。
下载完了之后我们打开anaconda prompt,输入命令进入下载好的压缩文件所在目录。比如,我下载在E盘里面,使用如下命令
E:
cd E:\chrome下载内容
conda install --offline torchvision-0.4.0-py37_cu100.tar.bz2
conda install --offline pytorch-1.0.0-py3.7_cuda100_cudnn7_1.tar.bz2
安装完成后就大功告成啦。
步骤四:测试pytorch是否安装成功
我们打开cmd输入如下命令
python
import torch
torch.__version__
如果不报错就表示pytorch环境安装成功啦。
PS:电脑里面有多个版本python,如何让cmd选择anaconda的python呢
先看一下我的环境变量设置
cmd在选择python解释器的时候会优先选择最上面那个,所以你只要把对应的python版本放在最上面就行啦。