PyTorch安装避坑

起因是更新torch版本后，运行程序会发现无法调用CUDA，走的时候CPU而非GPU。

import torch
print(torch.cuda.is_available())

运行以上代码会返回False，说明目前PyTorch无法调用CUDA。想起自己手贱更新过PyTorch应该就是因为这个了。

默认指令（如下）安装的是CPU版本PyTorch，要调用显卡核心必须安装对应版本PyTorch。

pip3 install torch torchvision torchaudio

安装前首先检查自己的CUDA版本，在命令行中输入下方命令。

nvidia-smi

返回结果中CUDA Version即CUDA版本号。如果没有安装过cuDNN需要根据另外教程安装，此处我以前安装过，故省略。

在PyTorch下载页（Start Locally | PyTorch）中选择正确选项，然后在你需要安装的环境中复制“Run this Command:”后方的内容并回车键执行。

如果此前安装过，请先进行卸载，否则如果版本号相同会认为已经安装过，

pip uninstall torch torchvision torchaudio