首页 > 解决方案 > “没有名为 torch 的模块”。但是在 Ubuntu 18.04.02 服务器版中安装了带有 conda 的 pytorch 1.3.0

问题描述

使用 conda 安装 pytorch:

(base) (3.8.0/envs/my_virtual_env-3.8.0) marco@pc:~/facenet_pytorch/examples$ conda install  
pytorch torchvision cpuonly -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done

# All requested packages already installed.

我更新了康达:

(base) (3.8.0/envs/my_virtual_env-3.8.0) marco@pc:~/facenet_pytorch/examples$ conda update 
conda
Collecting package metadata (current_repodata.json): done
Solving environment: done

# All requested packages already installed.

安装 mkl=2019 :

(base) (3.8.0/envs/my_virtual_env-3.8.0) marco@pc:~/facenet_pytorch/examples$ conda install  
mkl=2019
Collecting package metadata (current_repodata.json): done
Solving environment: done

# All requested packages already installed.

(base) (3.8.0/envs/my_virtual_env-3.8.0) marco@pc:~/facenet_pytorch/examples$ conda list | grep 
torch
cpuonly                   1.0                           0    pytorch
facenet-pytorch           0.1.0                    pypi_0    pypi
pytorch                   1.3.0               py3.7_cpu_0  [cpuonly]  pytorch
torchfile                 0.1.0                    pypi_0    pypi
torchvision               0.4.1                  py37_cpu  [cpuonly]  pytorch

但它仍然说“没有模块火炬”:

(base) (3.8.0/envs/my_virtual_env-3.8.0) marco@pc:~/facenet_pytorch/examples$ python3
Python 3.8.0 (default, Oct 30 2019, 16:20:23) 
[GCC 7.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'
>>> 

我发现问题只出现在 python 3.8.0 版本

(base) marco@pc:~/facenet_pytorch$ python3
Python 3.7.3 (default, Mar 27 2019, 22:11:17) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> 

Ubuntu 18.04.02 服务器版

或者,正如您所说,这只是 python 环境的问题。但我不明白为什么只激活 conda 环境,使用“conda activate”,它不起作用

马可

标签: python-3.xmodulepytorchcondaubuntu-18.04

解决方案


首先使用以下命令创建一个 Conda 环境:

conda create -n pytorch_env python=3 ( you can create with any python version )

使用以下方法激活环境:

conda activate pytorch_env

现在使用以下命令安装 PyTorch:

conda install pytorch-cpu torchvision -c pytorch

转到 python shell 并使用以下命令导入:

import torch

推荐阅读