记录一下遇到的问题。

最近在搭环境时出现了这个错误,网上找了许多方案都没法解决,说的大概都是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)

 

 

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐