五、Anconda下安装pytorch

五、Anconda下安装pytorch

（1）需要安装nvidia-driver
没有安装的可以看我的这篇文章：一、安装nvidia-driver
（2）需要安装cuda
没有安装的可以看我的这篇文章：二、安装cuda
（3）需要安装cudnn
没有安装的可以看我的这篇文章：三、安装cudnn
（4）需要安装anaconda
没有安装的可以看我的这篇文章：四、安装anaconda

1、创建虚拟环境

conda创建一个虚拟环境，用于使用pytorch，输入命令（-n后面为环境的名字，python版本指定最好不要小于3.7）：

conda create -n pytorchEnv python=3.8

成功截图：

2、切换环境

输入命令：

conda activate pytorchEnv

成功截图（此时命令行前为你激活的环境）：

3、选择pytorch安装包

网址：https://pytorch.org/get-started/locally/

根据系统和cuda版本选择要安装的pytorch：

4、运行安装命令

复制步骤三截图中的命令，输入：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

等待安装成功：

4、验证是否使用GPU

输入命令，进入python命令界面：

python

导入torch包：

import torch
print(torch.cuda.is_available())

运行截图如下：

当出现true时，代表安装成功