admin管理员组

文章数量:1530041

折腾了一周了吧，终于是在自己的电脑上编译完成，并且可以跑通Pytorch训练代码了。我也是小白一个，如果有专业词汇使用不当，还请见谅。
编译过程参考了官方编译文档, JC大神的博文以及github里面的一些Issue集合而成。向他们表示诚挚的感谢。

  • 简单介绍一下硬件：
    CPU ：I5 9400F
    GPU：AMD Radeon VII
    内存：64G

  • 软件环境：
    Ubuntu 16.04 LTS
    Anaconda3 (python=3.6)
    Rocm 3.0

  • 编译过程：

  1. 安装rocm，按照官方的文档安装即可
  2. 配置编译需要的环境，确保所需的包是安装了的。
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

本文标签： 实录深度GPUamdPytorch

版权声明：本文标题：AMD的GPU拿来跑深度学习？Rocm3.0&Pytorch@Ubuntu16编译实录 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/xitong/1726771092a1083882.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。