首先介绍环境: 

保证Cuda与Pytorch的版本对齐就可以了。

nvcc -V 查看原来装的是cuda11.3版本

去Pytorch官网找到相应指令下载即可:

Ctrl+F:cuda11.3 就在诸多版本中找到啦,一定找

torch的版本+cuda版本。我之前错误安装的torch,只会显示torch的版本,没有后面的+。

 

# CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

感谢这篇博客:

安装 真·GPU版 Pytorch,解决torch.cuda.is_available()输出False问题 - 知乎 (zhihu.com)

然后重新开一个终端(一定要重开一个),pip检查一下版本是否对。

然后  ↓↓↓检查一下就ok啦。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐