我是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