微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

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