2021-08-18

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

  • 一、前言
    • 本文以 windows10 为例子,其它例子都相似
  • 二、安装过程
    • 2.1 登录pytorch官网进行查找对应的下载命令
    • 2.2 在anaconda prompt环境中创建新的虚拟环境
    • 2.3 查看目前所有的虚拟环境、以及激活对应的虚拟环境
    • 2.3 环境验证
    • 2.4 如果要进入不同项目,可以选择退出虚拟环境
  • 三、总结

一、前言

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

二、安装过程

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

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

2021-08-18_第1张图片

  • 选择好之后,在绿色框位置就会出现一串的命令
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

2021-08-18_第2张图片

  • 创建成功

2021-08-18_第3张图片

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

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

2021-08-18_第4张图片

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

2021-08-18_第5张图片

  • 输入官网提供的命令行,在对应的虚拟环境进行下载pythrch
pip3 install torch torchvision torchaudio

2021-08-18_第6张图片

  • 因为用普通的镜像源下载会比较满,通过临时用豆瓣源的方法,2分钟安装成功。
pip3 install --index-url https://pypi.douban.com/simple  torch torchvision torchaudio

2021-08-18_第7张图片

2.3 环境验证

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

2021-08-18_第8张图片

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

deactivate  pytorch_env

三、总结

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

你可能感兴趣的:(pytorch安装,环境配置,pytorch,深度学习,机器学习,神经网络,anaconda)