python - 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

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

使用命令检查您的版本：

python --version

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

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

Answer 2

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

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 3

问题的另一个可能根源是运行 Pycharm 的 Flatpak 版本。我花了一些时间才弄清楚，但 Flatpak 不会加载用户特定的环境变量文件（即~/.profile，~/.bash_profile和 conda 环境）。我已经删除了 Flatpak 版本并安装了 snap 版本 ( sudo snap install [pycharm-professional|pycharm-community] --classic)，它加载了正确的 PATH 允许正确加载 CUDA。

Answer 4

我有一个类似的问题。

受@the-lay 回答的启发，打电话import os; os.system('')解决了我的问题。