首先介绍环境:
保证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啦。