如何解决从菜单模式和终端模式运行时,不同的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 举报,一经查实,本站将立刻删除。