理论上pytorch官网给的命令行下载命令会自行安装cuda和cudnn,一键安装!但是这是在你能翻墙,网速好的情况下。
我按照csdn的教程自己下了好几次,都卡了,没办法,只能按照步骤一步步装。
1. 安装cuda,我的版本是cuda9.0
2.安装与cuda版本对应的cudnn
(这两步的教程特别多,我这就不写啦)
3.装完cuda和cudnn之后,安装anaconda的Linux版本,官网下载即可。下载之后按照这个安装:https://blog.csdn.net/u013069552/article/details/78349716
(我这也有,需要的百度网盘:链接:https://pan.baidu.com/s/1HYDkQIoVrwzNSEUWDbHtXw
提取码:3zw7 )
4.下载安装Linux版本的pytorch,我下载的:版本0.4.1 - Linux - Python3.6(torch-0.4.1-cp36-cp36m-linux_x86_64)
(同样的,百度网盘链接:链接:https://pan.baidu.com/s/1uhKx63tYXnxgTZWSyVdZHw
提取码:jot9 )
pip install torch-0.4.1-cp36-cp36m-linux_x86_64.whl
之后再安装numpy(装了的就不用了)
5. 最后验证一下,输入以下命令:
import torch
torch.cuda.is_available()
如果不报错,且输出的是True,说明成功安装。