在这里插入图片描述

一、问题描述

使用gpu运行深度学习代码时,遇到这个错误。

二、错误原因:

与cuda驱动的版本有关。

三、解决办法:

1.打开命令行输入:nvidia-smi
2.查看CUDA版本
在这里插入图片描述
3.打开pytorch官网,下载对应CUDA版本的pytorch
在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐