- 问题:我在复现代码的时候,发现自己只拥有AMD的GPU,对于一个硬件小白来说,怎么办呢?我想看看怎么使用;
- 解决:
- 首先要安装支持AMD的GPU的
pytorch
,pytorch;
- 使程序在安装了支持AMD的GPU的pytorch的python环境中运行代码,下述代码可以检测在现有的环境中是否可以调用AMD的GPU:
# AMD
# use_roc = torch.cuda.is_available() # 对于 AMD GPU,ROCm会通过CUDA接口暴露
# print(f"是否可以使用 ROCm:{use_roc}")
use_roc = torch.version.hip is not None and torch.cuda.is_available()
print(torch.version.hip)
print(f"是否可以使用 ROCm:{use_roc}")
print(f"PyTorch 版本:{torch.__version__}")
print(f"CUDA 可用性:{torch.cuda.is_available()}")
print(f"HIP 版本:{torch.version.hip}")
输出:
嘿嘿,之前一直失败,直到切换了正确的python环境才成功,原来一直都没有使用安装了支持rocm版本的pytorch!!!小迷糊😊