这是一份非常简单快速的PyTorch安装方案,如果你已经安装了Anaconda,想要在Anaconda的基础上安装CPU版本的PyTorch,可以参考下文。下面的内容同样可以用来解决import torch出现错误ModuleNotFoundError: No module named 'torch'这一问题。
分为两个部分进行介绍,pytorch快速安装步骤和安装遇到困难的过程。
我把最简单的方法整理为如下一整套流程。
前提:你已经安装了Anaconda
鼠标右键单击电脑,属性->高级系统设置->环境变量->系统变量->Path->编辑
如果
\anaconda3;
\anaconda3\Scripts;
\anaconda3\Library\bin;
三条路径都存在,说明环境变量配置成功。
GPU版本的pytorch对规模非常大的神经网络的计算更友好,但是如果你目前只是最基本的一些使用,可以暂且安装CPU版本的,可以避免应对GPU版本选择、更新等问题。
从电脑左下角Windows当中打开Anaconda Navigator

在左侧栏选择环境Environments

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

在打开的命令行中输入
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple pip添加清华镜像的方法:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple conda添加清华镜像的方法:
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 可以用下面的方法来测试一下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'

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

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