☰

无法导入 pytorch

tag-icon • tag-icon python anaconda
无法导入 pytorch

我正在尝试导入 pytorch，但出现错误，提示模块不存在。我通过 anaconda 安装，文件夹“pytorch-0.1.12-py35_2cu80”存在于 anaconda3/pkgs 文件夹中。为什么它无法识别它的存在？我应该重命名此文件夹吗？

—里克

/home/rik/anaconda3/bin/python /home/rik/PycharmProjects/self_driving_car/self_driving.py
Traceback (most recent call last):
  File "/home/rik/PycharmProjects/self_driving_car/self_driving.py", line 18, in <module>
    import pytorch
ImportError: No module named 'pytorch'

Process finished with exit code 1

答案1

导入 pytorch 的命令是

import torch

不是

import pytorch

答案2

看来模块pytorch没有安装。

首先确保您使用的是 anacondapython而不是系统版本。

pytorch通过condado安装

conda install pytorch torchvision -c soumith

或者通过 pip

对于 python3

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
pip install torchvision

对于 python2

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 
pip install torchvision

如果您的机器上没有支持的 Nvidia GPU，请不要安装 CUDA 版本。

相关内容