Featured image of post 在Anaconda下安装Pytorch的教程

在Anaconda下安装Pytorch的教程

记录安装Pytorch的教程

安装步骤:

  1. 安装Anaconda
  2. 安装Pytorch
  3. 检验安装

安装Anaconda

官网下载

下载速度较慢,官网地址: Anaconda 选择 Products -> Anaconda Distribution

点击 Download,可以下载最新版本

安装Pytorch

1. 打开Anaconda Prompt

jc_1_1

2. 配置清华源(可省略)

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 show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这四行代码一行一行输入,这个配置好之后,以后再安装其他的软件如果要用到清华镜像源网站就不用了重新配置了

如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的conda源(一定要先恢复默认,再换另一个!!!) 切回默认源:

conda config --remove-key channels

3. 创建环境

(1)创建环境pytorch,我使用Python版本是3.10

conda create -n pytorch python=3.10

如需查看Python版本,打开命令行窗口,输入

python --version

提醒pytorch对python的版本有要求,具体可以查看官网https://pytorch.org/get-started/locally/#windows-python,目前windows下支持的python版本是3.8到3.11,python版本过低记得更新。

(2)查看环境是否安装成功

conda info --envs

如果出现了pytorch,就说明安装成功

(3)进入pytorch环境

输入activate pytorch,左边的环境就从base变成了pytorch环境。

activate pytorch

4. 安装Pytorch

在此我们需要安装pytorch, torchvision,torchaudio三个包。

(1)判断自己的判断自己电脑是否有GPU

  • 如果你的电脑有NVIDA显卡,那么按照下列步骤下载

  • 如果没有这个显卡,那么 就下载安装CPU版本的pytorch

(2)选择合适的pytorch版本并下载

  • 打开命令提示行,输入nvidia-smi,查看当前驱动的版本号,观察Driver Version的值是否大于400,如果小于请更新显卡驱动。
nvidia-smi

jc_1_3

查看你的CUDA Version,然后与pytorch下载页面的CUDA比较,你下载版本的应该小于你本身的CUDA Version。

  • 进入 pytorch官网,下滑找到这个页面

jc_1_4

  • 选择好版本之后复制图片中的代码,粘贴到Anaconda Prompt中进行下载,弹出提示,输入 y,即可完成安装。下载时间较长请耐心等待。
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

注意此时的环境是pytorch

检验安装

1.下载完毕后,进入pytorch虚拟环境,输入

conda list

查看有没有pytorch或者torch,有代表成功安装

jc_1_5

2.输入python回车,再输入

import torch
torch.cuda.is_available()

如果返回值是True,那就代表安装成功啦!

jc_1_6


以上教程参考此文章,并加入了一些我认为需要注意的点。

Licensed under CC BY-NC-SA 4.0