安装GPU版本的pytorch（官网安装pytorch只有cpu版本）

如何检查自己的pytorch是否cpu还是GPU格式的呢？
可输入以下命令：

print(torch.cuda.is_available()) # 判断torch是否可进行gpu计算
print(torch.cuda.device_count()) # 识别可使用的gpu个数

1.查找电脑是否有GPU，如果没有即使按照了GPU版本的pytorch也不能使用；
可在资源管理器中查找；

2.在cmd命令行中（ctrl+R）使用命令

 nvidia-smi 

检查电脑的Driver Version 和 CUDA Version，接着去官网上看看相匹配的gpu版本；
官网地址如下：https://pytorch.org/
（注意，此时不要直接下载，官网下载会得到cpu版本的）

3.在这个网站上找到合适的版本下载到本地；
https://download.pytorch.org/whl/torch_stable.html
其中，cu就是cuda，cu101就是CUDA10.1版本；
在本地命令行使用指令进行安装

pip install D:\DownloadData\MyDownload\torch-1.10.0+cu113-cp39-cp39-win_amd64.whl

4.之后可以再次进行检测，就会发现下载好gpu版本的pytorch啦！