torch.cuda.is_available()一直返回False

问题记录

torch.cuda.is_available()一直返回False

今天跑扩充之后的数据集，突然发现一个被我忽视的问题，就是我的程序一直使用cpu跑的（震惊！！！怪不得上周跑的如此之慢），发现原因是torch.cuda.is_available()一直返回False，导致device调用的一直是cpu。（明明之前重新下了一次torch，怎么还这样？）
然后：

说实话，感觉自己有点der……
记录一下解决方法，防止以后出现同样的情况。
解决方法很简单，就是重新下载gpu版的torch。
官网：https://pytorch.org/
点进去直接选择对应版本（我的是cuda11.6）的命令，命令行运行就OK了

but……
真的很慢，会超时，会下载失败，会一动不动……
so……
我们去官网下载安装包，至于版本号和上面的命令上写的版本对应找就ok啦~
下载whl文件：https://download.pytorch.org/whl/torch_stable.html

下载后的whl包放入想要使用的环境下的……\Lib\site-packages里面，在Anaconda Prompt中运行以下命令：

pip install 包的文件名

如：

pip install torch-1.12.1+cu116-cp37-cp37m-win_amd64.whl
pip install torchvision-0.13.1+cu116-cp37-cp37m-win_amd64.whl
pip install torchaudio-0.12.1+cu116-cp37-cp37m-win_amd64.whl

解决啦！

gpu下一个epoch16秒，真爽！（遥想我上周一个epoch2分40秒，每天像个苦逼似的傻等and 训练这么快又不能有合理的借口划水了）