AssertionError: CUDA unavailable, invalid device 0 requested
1.6.0版本的torch不支持低版本的cuda 10.1 所以检测不到gpu,安装兼容10.1低版本cuda的对应torch版本1.6.0+cu101
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步