240112问题记录:nvidia-smi找得到GPU但python找不到

发布时间:2024年01月17日

我在计算集群上运行一个需要GPU支持的python based软件时发现了这个问题,最终发现是因为pytorch版本问题,用了cpu版本的才会导致GPU支持跟不上。

通过conda list就可以看到自己的torch是不是cpu版本。是的话需要卸载重装,我用的conda。从官网复制命令下载,不要使用清华源。torch支持的cuda版本大于等于自己已有的cuda版本的情况都是可用的。

最后在python中查看cuda可用性,显示成功。

(代码就不贴了,这个问题比较常见网上应该到处都是解决方案与代码)

参考内容:

亲测有效解决torch.cuda.is_available()返回False的问题(分析+多种方案),点进不亏_torch.cuda.is_available()是false-CSDN博客

文章来源:https://blog.csdn.net/dstyaaaa123456/article/details/135552682
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。