问题描述
我在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 (将#修改为@)