1. 检测方法

在cmd中进入自己的Python环境，输入：

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

如果输出是True，那么恭喜你就是GPU版本的，可以略过此文；

如果输出是FALSE，那么你的torch版本就是CPU版，可以参考本文以下经验。

2. 检查电脑CUDA驱动版本

（1）右键点击NVIDIA，打开控制面板

（2）点击帮助中的系统信息

（3）点击组件，查看CUDA版本

如图所示，我的电脑目前CUDA版本就是11.4，但是在我想要安装对应的GPU torch时，发现目前的CUDA版本已经在11.6及以上了，于是决定先对CUDA版本进行更新。（注：CUDA版本可以向下兼容，例如12.0版本可以使用11.6/11.7版本）。

3. 更新CUDA驱动

（1）进入英伟达官方驱动

（2）依据相应的版本，寻找NVIDA驱动更新

可以在任务管理器中查看显卡，例如我这边显示PC显卡为1660Ti：

找对版本后，直接下载就OK

（3）安装

（4）检查是否升级成功

依据前面的办法打开控制面板查看

已经更新为CUDA 12.0版本，可以向下兼容哦~

4. 安装GPU版本的torch

（1）方法一：在线安装，直接在Pytorch官网下载版本：Pytorch

Run this Command

（2）方法二：离线安装，在此网站下载相应包：torch

找到自己需要的torch版本进行下载

例如：torch-1.13.0+cu117-cp37-cp37m-win_amd64.whl

代表的是torch 1.13.0 版本， CUDA 11.7 版本， Python 3.7 版本

下载后将包放入conda的env文件夹下

在cmd中进入该文件夹，进行安装

进行验证

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

如果显示torch版本显示+cu

以及 True

则安装完毕！