Pytorch官网:https://pytorch.org/
安装的时候如果选择在官网上下载安装的话下载会很慢,试过梯子也是很慢,此处提供一种快速安装的方法。
1.由于我是window系统,我选择了国内清华源下载,https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/,找到自己适配的版本将其下载下来
注意:py3.7是指你的python版本是python3.7;
cuda100是指cuda的版本是10.0版本。
如果不知道自己的电脑的cuda版本号是多少,可以在cmd输入以下命令查看:
nvcc -V
我的是python3.7,cuda是10.0,在清华源那里找的时候就找这个pytorch-1.0.1-py3.7_cuda100_cudnn7_1.tar.bz2。
2.下面是我用git命令行进行conda的安装(根据自己习惯来,这里用管理员身份打开Aaconda Prompt执行该命令也是可以的。。)
(1)进入下载刚才的文件的目录下
(2)执行以下命令:
conda install --use-local pytorch-1.0.1-py3.7_cuda100_cudnn7_1.tar.bz2
结果显示:
备注:如果遇到权限问题,就用管理员身份打开Aaconda Prompt执行该命令。