Pytorch简易安装

image from finaa

再次步骤前已经默认安装了anaconda(或miniconda),如果没安装请先安装

1.配置清华源

为了在国内能够快速安装 PyTorch, 需要先在 ~/.condarc 的设置清华源:

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
auto_activate_base: false
show_channel_urls: true

通过命令行的方式设置

这里我们只讲关于命令行设置的方式,通过cat直接读入的方式读者可以自行研究(在Linux/WSL/git bash中可以使用cat命令)

直接进入anaconda prompt后在base环境中输入一下代码:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
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/main/

conda config --set auto_activate_base false
conda config --set show_channel_urls true

运行结束后,可以通过:

conda config --show channels

检查是否配置成功

显示如下则配置成功:

2.创建新的虚拟环境

这里我们创建的是python=3.9版本的,基本使用是完全没问题的

conda create -n ‘your env name’ python=3.9

其中‘your env name’是你的虚拟环境名称,可以自己定义

等待加载完成之后,根据要求输入:

conda activate ‘your env name’

即可进入虚拟环境

3.安装PyTorch

⭐ 比较推荐去pytorch官网,自行选取合适的指令下载

或者按照一下操作进行:

这时候我们只需要在命令行中输入:

conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia

选择[y]后等待安装完成就好

4.测试是否安装成功

确保处于刚才创建的虚拟环境中,在命令行输入:

python

进入python的交互式解释器模式

之后在>>>后按顺序输入:

import torch
torch.cuda.is_available()

若显示true则说明pytorch安装成功

完结撒花🌼~