Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.

记录一下遇到的问题。

最近在搭环境时出现了这个错误，网上找了许多方案都没法解决，说的大概都是cuda版本和pytorch版本不兼容，我按照pytorch官网的指令安装了许多次都没用.

 

 我的cuda版本是11.3，按照此指令安装后跑程序一直报上述错误，看了下包后发现安装的cudatoolkit版本似乎是11.3.1，猜测是这个原因造成的，然后看见了

python中安装的pytorch对应的CUDA版本和服务器上的显卡驱动对应的CUDA版本不兼容，显卡驱动对应的CUDA版本应当>=pytorch对应的CUDA版本。

于是抱着试试的心态安装了cuda版本为11.1的pytorch

问题得到了解决。

参考：【报错解决】Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. - 知乎 (zhihu.com)