基于NVIDIA GPU安装pytorch（含Anaconda安装）

（一）安装Anacoda创建虚拟环境

（1）进入官网下载，官网链接：‘https://www.anaconda.com/’

（2）虚拟环境创建，在Anaconda Prompt 创建，输入指令：conda create -n 虚拟环境名字 python=版本，例如：conda create -n pytorch1 python=3.9

（3）拓展：删除创建的虚拟环境，输入指令conda deactivate先确保退出该环境，再输入conda env -n 虚拟环境名字

（二）GPU版本的-GPU与CUDA准备工作

（1）确定显卡型号

打开“任务管理器”，点击“性能”方可查看显卡型号。

（2）确定显卡算力

由显卡型号查看对应的显卡算力。

（3）确定CUDA Runtime

（4）确定驱动CUDA Driver Version

在Anaconda Prompt 输入指令：nvidia-smi，可查询。

结合（3）和（4）两部分CUDA Runtime的交集部分均支持。

（5）利用conda或pip安装pytorch

去pytorch官网查询安装命令（结合（4））

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

安装完成，本文到此结束