Não é possível importar o pytorch

Question

Estou tentando importar o pytorch, mas recebo um erro que o módulo não existe. Instalei via anaconda, e a pasta "pytorch-0.1.12-py35_2cu80" existe na pasta anaconda3 / pkgs. por que não reconhece que está aí? devo renomear esta pasta?

-Rik

/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

    

Answer 1

O comando para importar o pytorch é

import torch

não

import pytorch

    

Answer 2

Parece que o módulo pytorch não está instalado.

Primeiro, certifique-se de usar o python do anaconda e não o sistema um.

Para instalar pytorch via conda do

conda install pytorch torchvision -c soumith

ou via pip

Para python3

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

Para python2

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

não instale a versão CUDA, se você não tem Nvidia GPU em sua máquina que suporta.