Pytorch 无法在 PyCharm/IntelliJ 中与 CUDA 配合使用

Question

我刚刚通过 Anaconda 下载了带有 CUDA 的 PyTorch，当我在 Anaconda 终端中输入：

import torch
if torch.cuda.is_available():
    print('it works')

然后他输出；这意味着它有效并且可以与 PyTorch 一起使用。

但是当我进入 IDE（PyCharm 和 IntelliJ）并编写相同的代码时，它不会输出任何内容。

有人可以向我解释一下如何让它在 IDE 中运行吗？

Answer 1

这也让我发疯......最终帮助我的是第一个链接，它说使用 PyCharm“终端”运行 pip install 命令（来自 PyTorch 网站）。这解决了我所有的问题。（当时我已经安装了 pytorch 3 次并尝试了不同的解释器......）

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

我希望这可以帮助人们避免数小时的头痛。:)