当遇到`AssertionError: CUDA unavailable, invalid device 0 requested`的问题时，可以通过检查PyTorch的GPU可用性来诊断。如果`torch.cuda.is_available()`返回`False`，说明当前环境没有GPU支持。解决方法包括安装GPU版本的PyTorch，可以从PyTorch官网获取对应环境的pip指令进行在线安装，或者下载离线包进行安装。在安装完成后，验证`torch.cuda.is_available()`应返回`True`，问题即可得到解决。"
98367709,8669367,最小二乘法在线性代数中的应用解析,"['线性代数', '数值计算', '数学模型', '数据拟合', '矩阵理论']
进入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.html
https://download.pytorch.org/whl/torch_stable.html
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
最低0.47元/天 解锁文章
扫一扫
2万+
