Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
解决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
问题得到了解决。
更多推荐
已为社区贡献1条内容
所有评论(0)