我刚刚通过 Anaconda 下载了带有 CUDA 的 PyTorch,当我在 Anaconda 终端中输入:
import torch
if torch.cuda.is_available():
print('it works')
Run Code Online (Sandbox Code Playgroud)
然后他输出;这意味着它有效并且可以与 PyTorch 一起使用。
但是当我进入 IDE(PyCharm 和 IntelliJ)并编写相同的代码时,它不会输出任何内容。
有人可以向我解释一下如何让它在 IDE 中运行吗?