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 次并尝试了不同的解释器......）

https://www.datasciencelearner.com/how-to-install-pytorch-in-pycharm/

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

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

Answer 2

我还花了 10 多个小时试图找出问题所在。我最终检查了口译员是否有问题。 问题是 python 的版本（我有 v.3.8）。升级到最新版本后，问题解决了。

使用命令检查您的版本：

python --version

如果您的版本较旧（当前版本为 3.9.2），请从官方 python 站点安装最新版本： https://www.python.org/downloads/release/python-392/

安装后确保您使用较新的版本（使用上面的命令）

Answer 3

我也有类似的问题。

受到@the-lay答案的启发，致电

import os; os.system('')

解决了我的问题。

Answer 4

问题的另一个可能根源是运行 Pycharm 的 Flatpak 版本。我花了一些时间才弄清楚，但 Flatpak 不会加载用户特定的环境变量文件（即

~/.profile

、

~/.bash_profile

和 conda 环境）。我已经删除了 Flatpak 版本并安装了 snap 版本 (

sudo snap install [pycharm-professional|pycharm-community] --classic

)，它加载了正确的 PATH，允许正确加载 CUDA。

Answer 5

就我而言，这是我安装的 PyTorch 的特定版本。要检查 PyTorch 构建：

import torch
print(torch.__version__)

对我来说，它显示

2.0.1+cpu

，表示仅CPU版本。我按照官网推荐的方法重新安装了PyTorch。现在它显示

2.0.1+cu118

，表示 CUDA 兼容的构建。

---