AssertionError: CUDA unavailable, invalid device cuda 0 requested

先单独创建一个py文件用于测试cuda是否配置正确

print(torch.cuda.is_available())

运行后输出的是True，这代表你的cuda已经安装好了。
打开报错的程序，在主程序里面加入上面那一句：

print(torch.cuda.is_available())

然后把device改成cuda，默认是cuda0
然后运行问题就解决了。