Pytorch 不适用于 PyCharm/IntelliJ 中的 CUDA答案

【问题标题】：Pytorch doesn't work with CUDA in PyCharm/IntelliJPytorch 不适用于 PyCharm/IntelliJ 中的 CUDA
【发布时间】：2021-04-02 22:54:53
【问题描述】：

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

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

然后他输出；这意味着它可以工作并且可以与 PyTorch 一起工作。

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

有人可以向我解释一下我如何才能让它在 IDE 中工作吗？

【问题讨论】：

你真的安装了 cuda 吗？
是 11.0 版
Pytorch 在 pycharm 之外工作，所以这显然不是 cuda 或硬件的问题。您是否将 pycharm 项目配置为使用现有的 conda 环境？
你在这两种情况下使用相同的 conda 环境吗？您是否尝试过在 PyCharm 的内置终端中运行您的文件？
在 PyCharm 的终端中也不起作用

标签： python intellij-idea pycharm pytorch

【解决方案1】：

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

使用命令检查您的版本：

python --version

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

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

【讨论】：

【解决方案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

我希望这有助于节省几个小时的头痛。 :)

【讨论】：

【解决方案3】：

我也有类似的问题。

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

【讨论】：

【解决方案4】：

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

【讨论】：