Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)

一.准备anaconda

没有anaconda可以去官网下载一下,https://www.anaconda.com/,有人觉得官网下载过慢的话,这里给你提供一个镜像网址:https://mirrors.bfsu.edu.cn/anaconda/archive/
安装步骤:
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第1张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第2张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第3张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第4张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第5张图片
然后等待几分钟安装就行了
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第6张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第7张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第8张图片
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第9张图片
这样就安装好啦!

二.用anaconda安装pytorch

1.在开始程序中打开Anaconda Prompt
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第10张图片
进入界面
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第11张图片

2.在步骤3之前,为了防止anaconda安装pytorch过慢或者失败,这里先设置清华源,里面的命令要一行一行执行

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

这里所输入的命令是没有任何提示的
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第12张图片

3.通过conda create -n pytorch python=3.9命令创建pytorch环境,python的版本可以自由选择
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第13张图片

等待下载
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第14张图片
下载完毕
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第15张图片
4.通过conda activate pytorch命令进入pytorch环境,输入命令行后,发现括号里有所变化,说明进入成功
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第16张图片
5.下载pytorch
(1)首先进入pytorch官网:https://pytorch.org/,进入页面后,往下翻,找到这个。
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第17张图片

注意

我们要选择自己电脑对应的cuda版本,可以通过进入电脑的命令界面(window+R,然后输入cmd),使用nvidia-smi命令查看,有些人肯定会显示如下图的错误:
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第18张图片

报错原因

因为它找不到该命令。这个文件是一个exe文件,一般都在下面这个文件夹中。

C:\Program Files\NVIDIA Corporation\NVSMI

解决方案

1.添加环境变量:在path里面添加上面的路径,就可以访问到这个命令,就可以执行了。
2.更改命令路径:使用cd命令修改cmd中的文件夹,跳转到 ‘nvidia-smi’ 命令所在的文件夹。然后再次输入该命令即可。

cd C:\Program Files\NVIDIA Corporation\NVSMI

     nvidia-smi

执行之后可以看到如下信息:
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第19张图片
我们可以看到cuda的版本是11.0,所以上面我选择的10.2版本的,因为版本低一点是可以的,但不能选择高于11.0的版本。

要是显卡版本太低怎么办

解决:进入这个网址https://www.nvidia.com/download/index.aspx?lang=cn选择我们自己显卡对应的型号,然后下载进行升级。
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第20张图片
点击搜索,然后跳转到另一个页面进行下载
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第21张图片

怎么查看显卡的型号呢

解决

1.通过下载鲁大师等软件进行检测
2.通过进入任务管理器,点击性能选项,往下翻就可以看到

Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第22张图片
6.复制命令行,进行下载
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第23张图片

注意:这里复制的时候把 -c pytorch去掉,否则会用官网的通道,导致下载过慢

在pytorch环境下写入复制的命令后,输入y,然后等待下载安装
Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第24张图片
下载完成后,我们来检验一下是否成功

Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程)_第25张图片
显示输出为true,说明用anaconda安装pytorch成功了!

感谢大家的阅读,不足之后还望指教!

你可能感兴趣的:(python,pytorch,anaconda)