PyTorch快速安装技巧!ModuleNotFoundError: No module named 'torch'解决
小小筱筱_
编辑于 2022年06月05日 17:09

这是一份非常简单快速的PyTorch安装方案,如果你已经安装了Anaconda,想要在Anaconda的基础上安装CPU版本的PyTorch,可以参考下文。下面的内容同样可以用来解决import torch出现错误ModuleNotFoundError: No module named 'torch&#​39;这一问题。

分为两个部分进行介绍,pytorch快速安装步骤和安装遇到困难的过程。


一、PyTorch快速安装步骤

我把最简单的方法整理为如下一整套流程。

前提:你已经安装了Anaconda

1.检查Anaconda环境变量配置

鼠标右键单击电脑,属性->高级系统设置->环境变量->系统变量->Path->编辑

如果

\anaconda3;

\anaconda3\Scripts;

\anaconda3\Library\bin;

三条路径都存在,说明环境变量配置成功。

2.安装CPU版本的pytorch

GPU版本的pytorch对规模非常大的神经网络的计算更友好,但是如果你目前只是最基本的一些使用,可以暂且安装CPU版本的,可以避免应对GPU版本选择、更新等问题。

从电脑左下角Windows当中打开Anaconda Navigator

在左侧栏选择环境Environments

点击base(root)右边的绿色三角形,选择terminal

在打开的命令行中输入

代码块
JavaScript
自动换行
复制代码
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
复制成功

pip添加清华镜像的方法:

代码块
JavaScript
自动换行
复制代码
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple 
复制成功

conda添加清华镜像的方法:

代码块
JavaScript
自动换行
复制代码
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 
复制成功

3.测试pytorch是否安装成功

可以用下面的方法来测试一下pytorch是不是安装成功了。

4.在Jupyter book中导入pytorch

值得一提的是,后需要安装新的包时,也可以用同样的方法,从environment base进,然后输入命令语句。


二、安装过程和遇到的困难

下面简单讲一下我的安装过程。

首先是在conda命令行安装pytorch,由于没有使用镜像,安装会非常非常慢!小伙伴们可以不用尝试这种了,直接设置清华镜像安装会快很多很多。

安装好之后在命令行里测试了pytorch是完全没有问题的,也就是安装成功了。但是!在jupyter book中import torch会报错如下:

import pytorch

ModuleNotFoundError                       Traceback (most recent call last)<ipython-input-1-d35c46f8d1a2> in <module>----> 1 import pytorchModuleNotFoundError: No module named 'pytorch&#​39;

import torch

ModuleNotFoundError                       Traceback (most recent call last)<ipython-input-1-eb42ca6e4af3> in <module>----> 1 import torchModuleNotFoundError: No module named 'torch&#​39;

conda命令行import torch成功但在Jupyter book中导入失败

于是我就暂时放弃了Jupyter,想改用pycharm来import torch,仍然会报错,需要设置pycharm的解释器,尝试了各种方法之后,依然导入失败,所以最终采取了在navigator中open terminal进行安装的方法,这种应该是最快最简便的。之前导入不成功的可能原因是环境的选择和配置不对。


祝安装过程一切顺利。