0.背景
昨天看到新闻报道,
PyTorch 1.8来了!正式支持AMD GPU,炼丹不必NVIDIA
ROCm是AMD公司推出对标英伟达CUDA的计算库,这也就意味着AMD显卡在深度学习领域的生态更近一步,使用AMD显卡的小伙伴也可以用自己显卡跑深度学习了。
由于我的电脑一直是AMD显卡,所以前几年对ROCm一直有关注。之前ROCm对用户很不友好,尤其是对Pytorch,基本只能用docker才能跑的通。之前我也在AMD上安装过Caffe,排雷贴在这里:
昨天看到新闻报道,
PyTorch 1.8来了!正式支持AMD GPU,炼丹不必NVIDIA
ROCm是AMD公司推出对标英伟达CUDA的计算库,这也就意味着AMD显卡在深度学习领域的生态更近一步,使用AMD显卡的小伙伴也可以用自己显卡跑深度学习了。
由于我的电脑一直是AMD显卡,所以前几年对ROCm一直有关注。之前ROCm对用户很不友好,尤其是对Pytorch,基本只能用docker才能跑的通。之前我也在AMD上安装过Caffe,排雷贴在这里: