我刚刚通过Anaconda用CUDA下载了PyTorch，当我在Anaconda终端中输入时：
import torch
if torch.cuda.is_available():
print('it works')然后他输出那个;这意味着它工作了，并且它与PyTorch一起工作。
但是当我使用IDE（PyCharm和IntelliJ）编写相同的代码时，它不会输出任何内容。
有人能给我解释一下如何让这个在IDE中工作吗？
我刚刚通过Anaconda用CUDA下载了PyTorch，当我在Anaconda终端中输入时：
import torch
if torch.cuda.is_available():
print('it works')然后他输出那个;这意味着它工作了，并且它与PyTorch一起工作。
但是当我使用IDE（PyCharm和IntelliJ）编写相同的代码时，它不会输出任何内容。
有人能给我解释一下如何让这个在IDE中工作吗？
5条答案
按热度按时间i5desfxk1#
这也快把我逼疯了...最后帮我的是第一个链接，上面说使用PyCharm“终端”运行pip安装命令（来自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
我希望这有助于保存某人几个小时的头痛。：）
vql8enpb2#
我还花了10多个小时试图找出问题所在。最后我检查了解释器是否有问题。问题是python的版本（我有v. 3.8）。升级到最新版本后，问题解决了。
使用以下命令检查您的版本：
如果你的版本是旧的（当前版本是3.9.2），请从官方python站点安装最新版本：https://www.python.org/downloads/release/python-392/
安装后，请确保您使用较新的版本（使用上面的命令）
c6ubokkw3#
我也有类似的问题。
受@the-lay回答的启发，调用
import os; os.system('')解决了我的问题。wtlkbnrh4#
问题的另一个可能的根源是运行Pycharm的Flatpak版本。我花了一些时间才弄清楚，但Flatpak不加载用户特定的环境变量文件（即
~/.profile，~/.bash_profile和conda环境）。我已经删除了Flatpak版本并安装了一个snap版本（sudo snap install [pycharm-professional|pycharm-community] --classic），它加载了正确的PATH，允许正确加载CUDA。camsedfj5#
另外一个是entonys的回答：（谢谢你，恩托尼！）
这个命令在PyCharm中没有给我带来依赖性错误：
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio===0.13.0 -f https://download.pytorch.org/whl/torch_stable.html