从菜单模式和终端模式运行时,不同的CUDA结果

问题描述

我在vscode中有一个简单的test.py代码,如下所示:

import torch
# check if pytorch is using cuda/gpu
print("CUDA available: ",torch.cuda.is_available())

和以下settings.json

{
    "python.venvPath": "/Users/myname/Anaconda3/envs","python.pythonPath": "C:\\Users\\myname\\Anaconda3\\envs\\pytorch\\python.exe"
}

我已经在基本conda中安装了pytorch的cpu版本,并安装了一个名为“ pytorch”的虚拟环境,并安装了常规/ gpu版本的pytorch。

现在,当我使用python test.py从终端运行代码时,我得到了结果:

CUDA available:  False

如果我从菜单运行代码,或者使用开始调试(F5)或不进行调试(Ctrl + F5)进行运行,则结果都是:

CUDA available:  True

出什么问题了?以及如何在终端模式下使用CUDA?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)