CUDA driver initialization failed, you might not have a CUDA gpupytorch 解决方案
问题描述
在coding的时候我们经常在指定device的时候用这么一句代码:
device = 'cuda' if torch.cuda.is_available() else 'cpu'
但是有时候我们会发现device确实是放在了cpu上面,所以为了明确出错的原因,我们在shell里先import了torch,再执行torch.cuda.is_available()
,发现在返回False
结果之前给出了错误原因,其中部分内容就是我们在标题中写的。
问题原因
这种情况一般来说有两种原因
- gpu的计算能力过差
pytorch慢慢已经不支持cc(compute capability)小于3的gpu了,所以可以先查看一下自己的gpu的cc大小。但总的来说,这种情况出现的几率不大。 - cuda版本不兼容
这种情况通常是因为conda的虚拟环境中install的cudatoolkit版本高于gpu本身的cuda版本。在linux平台下,我们通过nvidia-smi命令查看gpu驱动的cuda版本,看是否低于我们在环境中安装的cudatoolkit版本。
解决方案
- 如果是因为平台gpu的cc过低导致的,那只能破费购进新卡或者降低torch版本了
- 一种方式是我们更新平台gpu的cuda版本,如果你是这台服务器的root用户,非常建议通过这种方式将gpu的驱动更新到最新版本。但如果你只是这台服务器的普通用户,我们则需要降我们的pytorch版本以满足需求的cudatoolkit版本小于等于平台gpu驱动。当然,如果版本降得很夸张,非常建议跟管理员去扯扯皮hhh 😄
参考
https://forums.developer.nvidia.com/t/k1000m-cuda-driver-initialization-failed-you-might-not-have-a-cuda-gpu/226120
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhiahahi
系列文章
更多
同类精品
更多
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
微信运动停用后别人还能看到步数吗
PHP中文网 07-22