现在公司网站重要吗_semseo是什么意思_优化大师是干什么的_免费b站推广网站短视频

• 问题：我在复现代码的时候，发现自己只拥有AMD的GPU，对于一个硬件小白来说，怎么办呢？我想看看怎么使用；
• 解决：

1. 首先要安装支持AMD的GPU的pytorch，pytorch；
   
2. 使程序在安装了支持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!!!小迷糊😊