关于conda下虚拟环境能否调用GPU的问题 | 您所在的位置:网站首页 › 为什么要安装cuda › 关于conda下虚拟环境能否调用GPU的问题 |
我是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 实验室设备网 版权所有 |