关于conda下虚拟环境能否调用GPU的问题 您所在的位置:网站首页 为什么要安装cuda 关于conda下虚拟环境能否调用GPU的问题

关于conda下虚拟环境能否调用GPU的问题

2024-01-10 18:24| 来源: 网络整理| 查看: 265

我是N卡用户,笔记本是RTX2060

为了在自行创建的虚拟环境里调用N卡真的是折磨我半天

原问题如下:

我在anaconda prompt里激活了一个自己创建的虚拟环境,然后进入python,再导入torch包,再运行torch.cuda.is_available()指令,为什么就成了false了呢???

(下图中的True实际是False,我这是调用成功后截的图)

 关键问题是,我在base环境里按上述步骤走下来,torch.cuda.is_available()指令给出的就是true,这是为什么捏?

经多方查证,其实是版本问题,具体怎么回事我也不知道(好像是base环境里cuda版本比较低?)

解决方案是什么捏?

首先我们把RTX2060的驱动更新到最新(反正越新越好,向下兼容的CUDA版本也更多)

详见你的GeForce Experience(当然网上还有从官网下的教程,如果你没有GF experience的话)

 然后就是在虚拟环境里安pytorch这一步了(假设之前你的CUDA什么的都安好了)

我的CUDA版本是11.4

进入官网

 我们在pytorch官网选10.2!是10.2哦,不要盲目上最新的,真的各种毛病(

然后把这指令复制到你的anaconda prompt里回车就完事了

这个时候运行torch.cuda.is_available()就是True了

总结一下,关键就是我装的CUDA11.4,但我在虚拟环境里选择安装兼容CUDA10.2的pytorch,由于高版本的CUDA兼容低版本的pytorch,所以能正常work



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有