一.来源

这个详细介绍请看此链接指向：pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch（CUDA 12.1）快速安装-腾讯云开发者社区-腾讯云

二.简单介绍

首先打开cmd

输入下边代码查看配置,查看你的cuda支持多少版本的

nvidia-smi

像我这个就是12.8的 

如果说前面的nvidia没安装的，请看开头连接，有详细教程。

三.安装

直接先在环境里面设置一个镜像源，

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

然后输入

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html

关于说这些版本具体在哪，去官方的地址寻找适合的版本就行，有很多版本之间会有冲突，所以不太好找合适的版本，

链接如下：PyTorch 

 

大概如下即可满足下载。