torch.cuda.is_available（）在（base）中为True，在其他conda env中为False

Question

我在Anaconda提示符下运行此代码，并返回True。

(base) C:\User
torch.cuda.is_available()
True

但是当我运行其他conda环境时，它根本不起作用。

(pytorch_project) C:\User
torch.cuda.is_available()
False

问题似乎是 torch.version.cuda 的结果不同。

（基本） torch.version.cuda = 10.1

（pytorch_project） torch.version.cuda = 10.2

但是我不知道为什么他们与众不同...

如何将 10.2 降低到 10.1 ，并使 is_available（）== True ？

---

这是我的信息。

 Windows 10 / nvidia-smi=425.31 / CUDA ver=10.1 / pytorch=1.4.0 / torchvision=0.5.0

conda list

Answer 1

尝试从pytorch_project环境中卸载PyTorch和Torchvision，然后以这种方式重新安装：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

或者，如果您更喜欢点子：

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

有关详细信息和更多选项，请参见https://pytorch.org/get-started/locally/。

Answer 2

谢谢您的回答和评论。 <3

我已经解决了问题。

我使用 Visual Studio代码作为开发人员环境工具，但是正如我上传的图片所示，conda list指向完全相同的目录，这意味着我没有这样做。不能真正激活我的环境。

应该类似于(pytorch) C:\User，但应该是（Power Shell）PS C:\User。

为解决这个问题，我去了Settings→Terminal > Integrated > Shell Args: Windows并用{p>

settings.json

参考：https://blog.lcarbon.idv.tw/vscode-設定-anaconda-路徑至-visual-studio-code-終端機中windows/

通过启动新终端，我正确地获得了"terminal.integrated.shellArgs.windows": ["-ExecutionPolicy", "ByPass", "-NoExit","-Command","& 'C:\\Users\\user\\miniconda3\\shell\\condabin\\conda-hook.ps1'" ] ，并看到我的(pytorch) PS C:\User正在使用torch=1.5.0。

然后我运行conda list来更改pytorch版本。

还有Voilà!!

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch