5分钟，教你如何轻松安装pytorch框架

一、前言

本文以 windows10 为例子，其它例子都相似

二、安装过程

2.1 登录pytorch官网进行查找对应的下载命令

• 在这里，大家点击以下链接，就就可以进入链接: https://pytorch.org/get-started/locally/

• 选择好之后，在绿色框位置就会出现一串的命令

pip3 install torch torchvision torchaudio

2.2 在anaconda prompt环境中创建新的虚拟环境

• 本次默认已经安装好了anaconda了，如果大家还没有安装，可以参考一下链接。链接: https://blog.csdn.net/qq_44810972/article/details/108323498.
• 创建新的虚拟环境，不同项目有自己的虚拟环境，有的项目可能需要的编译器版本不同，这样就能有效防止互相干扰了。
• pytorch_env就是虚拟环境的名称
• python=3.7 就是选择对应的编译器

conda create -n pytorch_env python=3.7

• 创建成功

2.3 查看目前所有的虚拟环境、以及激活对应的虚拟环境

• 查看目前的所有虚拟环境

conda env list

• 激活虚拟环境(activate 环境名)

activate pytorch_env

• 输入官网提供的命令行，在对应的虚拟环境进行下载pythrch

pip3 install torch torchvision torchaudio

• 因为用普通的镜像源下载会比较满，通过临时用豆瓣源的方法，2分钟安装成功。

pip3 install --index-url https://pypi.douban.com/simple  torch torchvision torchaudio

2.3 环境验证

• 进入python 环境，输入import torch 没有报错即安装成功

2.4 如果要进入不同项目，可以选择退出虚拟环境

deactivate  pytorch_env

三、总结

• 本次记录了我在笔记本电脑安装pytorch的全过程，希望对大家有所帮助.