本文记录了在Ubuntu 16.04 LTS上，使用AMD Radeon VII GPU，通过Rocm 3.0和Pytorch进行深度学习的编译过程。内容包括安装ROCM、配置环境、修改Pytorch源码、解决编译问题及测试。在编译过程中，特别提到了对CUDA函数的Hipify处理，以及针对不同GPU型号的配置。此外，文章还给出了遇到问题时的排查建议，如更换源、应用补丁、设置环境变量等。
折腾了一周了吧，终于是在自己的电脑上编译完成，并且可以跑通Pytorch训练代码了。我也是小白一个，如果有专业词汇使用不当，还请见谅。
编译过程参考了官方编译文档, JC大神的博文以及github里面的一些Issue集合而成。向他们表示诚挚的感谢。
-
简单介绍一下硬件：
CPU ：I5 9400F
GPU：AMD Radeon VII
内存：64G -
软件环境：
Ubuntu 16.04 LTS
Anaconda3 (python=3.6)
Rocm 3.0 -
编译过程：
- 安装rocm，按照官方的文档安装即可
- 配置编译需要的环境，确保所需的包是安装了的。
sudo apt update
sudo apt install rock-dkms rocm-dev rocm-libs miopen-hip miopengemm hipsparse rccl rocthrust hipcub roctracer-dev
sudo apt install git python-pip
最低0.47元/天 解锁文章
扫一扫
2364
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
