Improved way to install tensorflow-rocm

Introduction

AMDGPU Radeon製品でTensorFlowを動かす方法まとめ。

Installation

基本ソフトウェアをインストール

1
2
sudo apt update
sudo apt -y install software-properties-common curl wget # for add-apt-repository

Python3.5.2をインストール

Python3.6/Python3.7は不安定要素があるので3.5.2がおすすめ。
Ubuntu18にインストールする場合は、Python3.6がベースなので、3.5.2にするときも以下の方法が参考になります。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
PYTHON35=false
if [[ `python3 --version` == *"3.5"* ]] ; then
echo 'python3.5 -- yes'
PYTHON35=true
else
echo 'python3.5 -- no'
PYTHON35=false
fi

if [ $PYTHON35 == 'true' ] ; then
sudo apt install -y python3.5 python3.5-dev python3-pip
else
sudo add-apt-repository -y ppa:deadsnakes/ppa
sudo apt-get update
sudo apt install -y python3.5 python3.5-dev python3-pip
sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.5 1
sudo update-alternatives --set python3 /usr/bin/python3.5
python3 --version
curl https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py
sudo -H python3 /tmp/get-pip.py --force-reinstall
#sudo apt-get remove -y --purge python3-apt
fi

AMDGPU Radeon用のGPUドライバのインストール

以前より簡単になっています。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
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 rocm-dkms rocm-libs miopen-hip cxlactivitylogger libnuma-dev rocm-smi
sudo usermod -a -G video $LOGNAME
/opt/rocm/opencl/bin/x86_64/clinfo

echo 'export ROCM_HOME=/opt/rocm' >> ~/.profile
echo 'export HCC_HOME=$ROCM_HOME/hcc' >> ~/.profile
echo 'export HIP_PATH=$ROCM_HOME/hip' >> ~/.profile
echo 'export PATH=/usr/local/bin:$HCC_HOME/bin:$HIP_PATH/bin:$ROCM_HOME/bin:$PATH:/opt/rocm/opencl/bin/x86_64' >> ~/.profile
echo 'export LD_LIBRARY=$LD_LIBRARY:/opt/rocm/opencl/lib/x86_64' >> ~/.profile
echo 'export LC_ALL="en_US.UTF-8"' >> ~/.profile
echo 'export LC_CTYPE="en_US.UTF-8"' >> ~/.profile

TensorFlow-ROCmのインストール

現在は、TensorFlow1.12.0が最新になっており、Kerasの組み合わせが良いバージョンは、
Keras2.2.2になりますので、Keras2.2.2を指定してインストールします。

1
2
3
sudo pip3 uninstall -y tensorflow
sudo pip3 install --user tensorflow-rocm
sudo pip3 install --user Keras==2.2.2

ソースからインストールする方法は以下に纏めてありますのでご参照ください。

https://github.com/aieater/rocm_tensorflow_info

以上の方法を以下の一行で、実行できます。

1
curl -sL http://install.aieater.com/setup_rocm_tensorflow_p35 | bash -

References


GPU EATER - AMD GPU-based Deep Learning Cloud