ROCm与tensorflow在ubuntu 18.04下的安装教程

最新推荐文章于 2024-08-10 07:35:17 发布

置顶

这个昵称好啊

最新推荐文章于 2024-08-10 07:35:17 发布

阅读量1.1w

点赞数 8

分类专栏： ROCm tensorflow 文章标签： ROCm tensorflow AMD ubuntu18.04

本文链接：https://blog.csdn.net/Lord_is_my_shepherd/article/details/92732210

版权

本文详述了在Ubuntu 18.04上安装ROCm 2.5和TensorFlow 1.13的步骤，包括计算机配置环境、ROCm介绍、安装教程、卸载方法及TensorFlow安装验证。ROCm是AMD的开放计算平台，类似NVIDIA的CUDA，支持深度学习框架如TensorFlow。

摘要由CSDN通过智能技术生成

ROCm与tensorflow在ubuntu 18.04下的安装教程

本来准备买Nvidai的显卡，可是囊中羞涩再加上听说AMD的卡也能使用tensorflow进行深度学习加速了，所以1149京东买了个蓝宝石RX580 8G 白金版。网上找安装教程时发现教程非常少，所以决定写一篇详细的安装攻略，以便下次重装系统时少走弯路。

一、计算机配置环境

CPU： intel i7 8700
GPU： AMD RX580 8G
内存： 8G*2 DDR4 2400MHZ
硬盘： SSD 120G
主板：华硕ROG B360
电源：安钛克550W
操作系统： Ubuntu 18.04 (不要先装Anaconda，装了也没关系解决办法，
解决办法点击跳转)
安装ROCm版本：最新2.5（截止2019.06.10）
tensorflow版本：1.13

二、ROCm介绍

github地址，
社区地址，相关blog

ROC全称为Radeon Open Compute，即为“Radeon开放计算”，其中Radeon为AMD GPU的一个系列。
ROCm 全称ROC platforM，是基于一系列开源项目的AMD GPU计算生态平台。类似于NVIDIA家的CUDA。截止2019.06，ROCm版本已经发布到2.5,支持的深度学习框架有Tensorflow、Caffe、Pytorch。
MIOpen：AMD发布的深度学习加速库，相当于NVIDIA家的cuDNN深度学习加速库，其实是根据cuDNN移植过来的。