pytorch是非常脆弱的,它的版本和cuda对应好才可以使用,所以cuda并不是越新越好的。pytorch其实是非常好下载安装的,难题就是cuda版本选择的问题。
1 打开Pytorch官网,选择previous-versions,按照要求选择自己的合适版本。注意系统的选择,有英伟达显卡的选择cuda,没有的选择cpu。安装之前看一下自己电脑显卡最高能支持什么版本的cuda!!!!
官网:PyTorch
2 对了,如果有的同学没有换源的先进行换源!!!点这里CSDN
打开Anaconda Promp,进入虚拟环境,输入官网上给出的conda代码。
3 安装测试
python
import torch
print(torch.cuda.is_available())