AMD GPU(A卡)+Tensorflow+Anaconda+ubuntu18.04.2 安装方法

  • AMD GPU(A卡)+Tensorflow+Anaconda+ubuntu18.04.2 安装方法
    • 安装Anaconda
    • 安装ROCm
    • 安装TensorFlow
    • PS:
    • 参考

AMD GPU(A卡)+Tensorflow+Anaconda+ubuntu18.04.2 安装方法

Linux下可以可以通过AMD的ROCm平台实现Tensorflow的加速,下面介绍详细的方法。
首先说明系统,本人尝试过Ubuntu16.04.6和18.04.2,16.04总在安装过程中卡在设置rocm-profiler这一步。建议使用18.04系统,如果有大神在16.04系统上安装成功,请告诉我,感谢。
因为本人习惯了使用Anaconda,故将其一起安装。如果不想使用conda后面也会给出使用系统内python3安装的方法。

安装Anaconda

下载Anaconda,目前ROCm版的Tensorflow已经支持到python3.6,所以下载Anaconda3-5.2.0即可,地址:
https://repo.continuum.io/archive/
下载后是一个.sh文件,直接运行安装即可。

安装ROCm

添加ROCm的仓库

wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main > /etc/apt/sources.list.d/rocm.list'

然后运行:

sudo apt update
sudo apt install rocm-dkms

安装TensorFlow

安装相关软件包

本人当时安装了以下软件包:

sudo apt update
sudo apt rocm-dev rocm-libs rocm-device-libs \
hsa-ext-rocr-dev hsakmt-roct-dev hsa-rocr-dev \
rocm-opencl rocm-opencl-dev rocm-utils rocm-profiler \
cxlactivitylogger miopen-hip miopengemm

官网最新给的方法是安装下面的软件包,大家可以试一下(官网网址:https://rocm.github.io/tensorflow.html):

sudo apt update
sudo apt install rocm-libs miopen-hip cxlactivitylogger

然后安装Tensorflow

pip install --user tensorflow-rocm

如果安装过程中提示有缺少的软件包,安装相应的包即可。

现在可以打开Anaconda,import tensorflow试一下了。
安装完后打开Anaconda的spyder可能会出现段错误,如果出现,安装一个额外的软包即可:

pip install pyopengl

如果不使用Anaconda,去掉安装Anaconda的步骤,然后安装pip:

sudo apt install python3-pip

然后:

sudo pip3 install --user tensorflow-rocm

即可。

PS:

ROCm平台只支持AMD4系以后的显卡,如:RX480,RX460。

参考

[1] TensorFlow通过AMD GPU加速(ROCm/Ubuntu 18.04)https://www.jianshu.com/p/6fabdcde831b
[2] Ubuntu+anaconda3+python3:运行spyder出现:段错误 (核心已转储) https://blog.csdn.net/sinat_41789705/article/details/82764155[]
[3] ROCm官网 https://rocm.github.io/dl.html

AMD GPU(A卡)+Tensorflow+Anaconda+ubuntu18.04.2 安装方法相关推荐

  1. ubuntu18.04下安装GPU实例

    ubuntu18.04下安装GPU实例 以下是用到的命令去安装GPU 测试显卡驱动是否安装成功 如何改变文本的样式 插入链接与图片 如何插入一段漂亮的代码片 生成一个适合你的列表 创建一个表格 设定内 ...

  2. ubuntu18.04上安装anaconda-python深度学习环境

    Anaconda是一套Python的发行版,发行版集成了必要的库,使用户可以一次性完成安装.Anaconda是一个侧重于数据分析的发行版,一些有助于数据分析的库,比如Numpy,Matplotlib, ...

  3. 在ubuntu18.04下安装标签软件labelme

    在ubuntu18.04下安装标签软件labelme 在论坛里面看到很多前辈在ubuntu16.04下安装软件labelme,索性模仿他们尝试了一下,看了很多的博客和论坛,尝试了很多的方法,最终仍然与 ...

  4. 双系统windows+ubuntu18.04的安装和基本配置

    一.安装前确认信息 1.确认引导模式 win+R 输入msinfo32,打开系统信息,可以看到BIOS模式为UEFI,表示 UEFI Boot 引导模式. 二.双系统制作流程 1.下载Ubuntu镜像 ...

  5. ubuntu18.04上安装ffmpeg

    ubuntu18.04上安装ffmpeg 下载ffmpeg wget https://johnvansickle.com/ffmpeg/builds/ffmpeg-git-amd64-static.t ...

  6. ubuntu18.04.4 安装 dlib (解决 Running setup.py install for dlib ... error错误)

    ubuntu18.04.4 安装 dlib (解决 Running setup.py install for dlib - error错误) 直接安装出现错误: Running setup.py in ...

  7. ubuntu18.04.4 安装百度硬盘

    文章目录 ubuntu18.04.4 安装百度硬盘 下载 打开网址 ubuntu18.04.4 安装百度硬盘 下载 打开网址 https://pan.baidu.com/download

  8. ubuntu18.04.4 安装 qq x64 版本

    文章目录 ubuntu18.04.4 安装 qq x64 版本 确定版本 根据你所使用的linux发行版选择格式: 1 代码安装 1 下载 2 安装qq 3 删除qq 2 图形化界面安装 1 下载 2 ...

  9. ubuntu18.04.4 安装 微信

    ubuntu18.04.4 安装 微信

最新文章

  1. 深度学习框架哪家强?国产框架OneFlow做了一份测评报告
  2. CentOS7 编译安装 Mariadb
  3. 被墙怎么搭建安卓开发环境
  4. Servlet: server applet与执行原理
  5. 扩展坞可以把手机投到显示器吗_你的电脑可以省了!华为Mate 30搭配绿联新一代扩展坞轻松变电脑...
  6. mac上安装MySQL
  7. java中equals的理解(1)
  8. 安装EPP的调试Zend Debugger
  9. sql视图能使用触发器吗_白刚玉粉末能重复使用吗
  10. [沈航软工教学] 团队项目地址汇总
  11. java注解接收上传文件_SpringMVC 处理 multipart/data 请求实现文件上传
  12. CF1040B Shashlik Cooking
  13. 2017百度之星资格赛:1004. 度度熊的午饭时光(01背包+记录路径)
  14. WinZip 11 汉化注册版
  15. python阿拉伯数字转换为英文,python – 将东部阿拉伯数字转换为西方阿拉伯数字...
  16. cents 7.0命令
  17. 写好一份数据分析报告需要注意的13个要点
  18. 数据结构几类排序的总结和完整代码 待续。。
  19. Amazon 4.7 星评,领域新经典,了解服务设计就读它
  20. MGR 8.0 + ProxySQL 2.0 部署实录

热门文章

  1. Vscode大纲视图查看代码结构,只显示类名和函数名,隐藏变量名的方式
  2. Mybatis模糊查询
  3. 程序员内功心法之函数栈帧的创建和销毁
  4. nginx+uwsgi负载均衡部署django项目
  5. python正则入门
  6. 静态数据成员友元函数
  7. QT tableWidget给单个表格添加背景颜色
  8. 计算机安全防护厂商,全球安全行业共抗暴雷 中国厂商表现抢眼
  9. Vue常见问题汇总及解决方案
  10. [oeasy]python0120_英语的崛起_英文字符_小写字母的由来_不列颠帝国