RuntimeError: No HIP GPUs are available

错误原因:

代码使用了GPU加速的torch,但是我下载的是C++版本的torch。使用pip list发现包如下:
请添加图片描述

参考HIP网站:
https://pytorch.org/docs/stable/notes/hip.html

解决方案:

卸载torch和torchvision,然后使用pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html安装即可。


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部