Pytorch部分报错问题

一、存在问题

1.链接库报错

问题报错：undefined symbol: __nvJitLinkAddData_12_1, version libnvJitLink.so.12

2.GPU不能使用

问题报错：在torch安装后测试gpu是否可用

torch.cuda.is_available()
false

返回false

问题分析：

可能是直接安装环境时，torch版本和机器上本身安装的cuda版本不符合，或者torch没有完全安装导致

解决方法：

比较暴力的方法是直接重装torch可以解决大部分问题，直接去pytorch官网，安装对应版本的torch，建议使用conda安装，可以直接安装，无需卸载操作。Start Locally | PyTorch

如果是nvidia-smi中，cuda版本为12.0，可以选择12.X版本，都可以使用

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia