AssertionError: CUDA unavailable, invalid device 0 requested问题

进入python查看当前pytorch是否可用gpu，发现是false

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

查看torch版本，发现是cpu，可能是安装的时候没有留意

>>> torch.__version__
'1.10.2+cpu'

 解决办法自然是安装GPU版本的torch,要么自动安装，可以进入pytorch官网，，会让你选向安装的环境，选好后帮你生成pip指令，直接安装即可，但是torch太大了，如果安装不成功，可以考虑离线安装，torch离线安装网站：https://download.pytorch.org/whl/torch_stable.htmlhttps://download.pytorch.org/whl/torch_stable.html

 下载对应的torch和torchvision版本，然后把下好的文件移动到随便一个目录，在该目录安装，比如：

(base) PS E:\cu101> pip install .\torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl .\torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
 

装好之后，可以验证一下： 

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

应该就解决了