问题描述
在coding的时候我们经常在指定device的时候用这么一句代码:文章来源:https://www.toymoban.com/news/detail-527330.html
device = 'cuda' if torch.cuda.is_available() else 'cpu'
但是有时候我们会发现device确实是放在了cpu上面,所以为了明确出错的原因,我们在shell里先import了torch,再执行torch.cuda.is_available()
,发现在返回False
结果之前给出了错误原因,其中部分内容就是我们在标题中写的。文章来源地址https://www.toymoban.com/news/detail-527330.html
到了这里,关于【CUDA driver initialization failed, you might not have a CUDA gpu】pytorch 解决方案的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!