7 votes

Ne peut pas importer pytorch

Je cherche à importer pytorch mais j'obtiens une erreur indiquant que 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 le reconnaît-il pas? Est-ce que je devrais 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

16voto

Rik Points 419

La commande pour importer pytorch est

import torch

et non

import pytorch

1voto

Arduino_Sentinel Points 1707

Il semble que le module pytorch ne soit pas installé.

Assurez-vous d'abord d'utiliser le python d'Anaconda et non celui du système.

Pour installer pytorch via conda, 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

Ne pas installer la version CUDA si vous n'avez pas de GPU Nvidia sur votre machine qui le supporte

SistemesEz.com

SystemesEZ est une communauté de sysadmins où vous pouvez résoudre vos problèmes et vos doutes. Vous pouvez consulter les questions des autres sysadmins, poser vos propres questions ou résoudre celles des autres.

Powered by:

X