Pytorch-GPU环境配置

Pytorch-GPU环境配置

一、基本步骤（已安装好Anaconda）

1、确认是否是英伟达显卡，确定好显卡型号。
打开NVIDIA控制面板，


2、上NVIDIA官网上下载对应版本的CUDA，安装CUDA和cuDNN，参考
https://blog.csdn.net/Angeldream123/article/details/123526012?spm=1001.2014.3001.5506
3、做完准备工作，开始创建虚拟环境

4、进入终端，打开创建的虚拟环境

5、在对应环境中安装gpu版本的torch，上Pytorch官方网站

复制命令到终端，开始安装即可。
6、检测
安装完成后，进入python环境进行检验。

注意最后返回结果为True即为安装成功。

二、错误总结

1、安装CUDA时，选用精简版，安装在C盘占地太多。然后在卸载时，卸载过多。
这里有一个小工具，DUU，可以将其卸载干净，但有时有些不必要卸载的也卸载了。如果发现驱动丢失，可以下载驱动精灵自查一下，重新安装即可。
2、torch.cuda.is_available()返回False
原因：按照官网的命令默认安装，但是我的torch却是CPU版本的。
在这里教大家一个自查的方法，进入Python输入

import torch
torch.__version__

如果返回是CPU版的torch，那么恭喜你也中招了。我也是查了好多网上的帖子，但是和我的原因不太一样。最后我本来打算手动安装，进入官网，找到对应的CUDA版本的torch，
https://download.pytorch.org/whl/torch_stable.html
但是我发现找不到python3.6版本的（我的虚拟环境是python3.6版本的），CUDA11.6最低是py3.7，我想这可能就是我装不上GPU版的torch的原因吧。
接着我将原本安装的torch等包卸载，重新选择安装11.3版本的

经过漫长的等待，终于安装好了。
重新测试

Ture！！！！终于，安装了一天终于好了。