web-dev-qa-db-fra.com

Impossible d'importer du pytorch

J'essaie d'importer pytorch, mais j'obtiens une erreur: le module n'existe pas. J'ai installé via anaconda, et le dossier "pytorch-0.1.12-py35_2cu80" existe dans le dossier anaconda3/pkgs. pourquoi ne reconnaît-il pas qu'il est là? dois-je renommer ce dossier?

-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
pythonanaconda
6
Rik

La commande pour importer pytorch est

import torch

ne pas

import pytorch
12
Rik

Il semble que le module pytorchne soit pas installé.

Tout d’abord, assurez-vous d’utiliser le nom pythond’anaconda et non celui du système.

Pour installer pytorchvia condaname__, faites

conda install pytorch torchvision -c soumith

ou via pip

Pour python3

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

Pour python2

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

n’installez pas la version CUDA si vous n’avez pas le GPU Nvidia sur votre ordinateur qui le supporte.

1
Arduino_Sentinel