Anaconda安装pytorch，配置gpu

目录

Anaconda的安装

Pytorch的安装

GPU的配置

---

Anaconda的安装

进入官网，点击Download

下好右键安装。一般只需要安装自己用户即可，选just me,next

选择安装路径,next

建议四个全选，免去手动添加环境变量的麻烦。Install

Pytorch的安装

开始菜单，找到新加的Anaconda文件，打开Anaconda Prompt

在打开的终端窗口，输入如下指令，创建名为pytorch的虚拟环境，其中3.xx为python解释器版本

conda create -n pytorch python=3.11

激活对应环境

activate pytorch

进入pytorch官网

复制下面指令到终端窗口并运行

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

依次输入

python
import torch
torch.cuda.is_available()

输出True说明安装正确无误

GPU的配置

配置gpu版本的pytorch，我们需要再外部配置好CUDA的显卡驱动，在虚拟环境内配置cudatoolkit，cudnn。

显卡驱动根据自己电脑显卡型号确定

官方驱动驱动下载并安装

cudatoolkit已经在pytorch安装时安装

还剩下cudnn，pytorch环境的终端输入命令：

pip install cudnn

至此，大功告成！