网上很多教程都是直接用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