首页 > 其他 > 详细

TensorFlow通过AMD GPU加速(ROCm/elementary OS 5.0)

时间:2020-06-30 13:37:14      阅读:65      评论:0      收藏:0      [点我收藏+]

https://www.jianshu.com/p/ff66c5dac823

elementary OS 5.0是基于Ubuntu 18.04制作,所以本流程也适用于Ubuntu 18.04。

ROCm升级1.91之后不需要安装AMD的GPU驱动程序,如果安装可能会出现问题。

安装elementary OS 5 之后的一些配置

详细配置请参阅:elementary OS 5.0安装后配置

安装软件更新管理器

sudo apt update
sudo apt install software-properties-common software-properties-gtk

运行软件更新管理器然后选择合适的镜像,我使用的是阿里云。

sudo software-properties-gtk

更新系统

sudo apt update
sudo apt upgrade

安装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 -y   build-essential   clang   clang-format   clang-tidy   cmake   cmake-qt-gui   ssh   curl   apt-utils   pkg-config   g++-multilib   git   libunwind-dev   libfftw3-dev   libelf-dev   libncurses5-dev   libpthread-stubs0-dev   vim   gfortran   libboost-program-options-dev   libssl-dev   libboost-dev   libboost-system-dev   libboost-filesystem-dev   rpm   wget 

安装ROCm驱动

sudo apt update &&     sudo apt install -y --allow-unauthenticated     rocm-dkms 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

添加用户

sudo adduser $LOGNAME video
sudo reboot

安装miniconda

在使用python时我们很多时候会用到conda,但是conda会改变python的一些配置。所以我们先安装conda,这样就不会出现在使用tensorflow时需要使用/usr/bin/python3这样的命令。
下载地址(64位):https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

cd 下载目录
chmod +x ./Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh

安装之后我们发现conda的python是3.7,而ROCm版的tensorflow支持3.5和3.6,所以我们需要将python降为3.6。

conda install python=3.6

安装python相关的软件包

sudo apt update && sudo apt install -y     python3-numpy     python3-dev     python3-wheel     python3-mock     python3-future     python3-pip     python3-yaml     python3-setuptools

安装TensorFlow

pip3 install --user tensorflow-rocm


作者:王胖BigWang
链接:https://www.jianshu.com/p/ff66c5dac823
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

TensorFlow通过AMD GPU加速(ROCm/elementary OS 5.0)

原文:https://www.cnblogs.com/dhcn/p/13213049.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!