注意:该方法默认已经提前创建好了虚拟环境!


1.安装 Conda( Miniconda 、Anaconda都可以)

  • 以下默认的虚拟环境是 python3.8 ,本文的 pytorch 以 CPU 版本为例(包括部分 GPU 版本的教程)。

2.GPU 版本需要确定 CUDA 版本( CPU 版本请跳过本节)

  • 在桌面点击右键,选择“NVIDIA控制面板”

  • 依次点击:帮助-->系统信息

  • 点击“组件”,如图所示,本电脑CUDA是11.4的


3.下载 Pytorch 离线包

  • 离线安装包推荐网站:https://download.pytorch.org/whl/torch_stable.html
  • 技巧:可以打开 pytorch 的官网(PyTorch)查看自己适合什么版本( CUDA 版本:如果通过第 2 部,发现自己的 CUDA 版本不对应,需要及时去更新),下图是 CPU 版本的示例(选择 None )然后再进入上面的网站单独下载,如:

  • 由于之前创建的虚拟环境是使用的 python3.8 ,所以此处对应的分别下载(cu开头的是GPU版本。中间“%2B是beta版本,实际可以下载正式版本。“cp 后面的数字代表 python 版本。“win_amd64表示适配的操作系统):

  • 最后下载得到的有:


3.安装

  • 首先打开cmd
  • 使用“conda activate xxxxx(之前创建的环境名字)”激活到当前环境
  • 再使用cd定位到下载torch的目录中
  • 最后依次pip install torch、torchvison、torchaudio(一定要输入完整的名字)
  • 全部安装完了试一试输入python进入解释器,再输入import torch。如果没有报错即安装成功。
  • 附:安装过程中可能有些包的下载速度有些慢,可以使用清华源替换.condarc这个文件,详细参考清华源的界面介绍

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐