POSTS
改良版tensorflow-rocmインストール方法
Introduction
AMDGPU Radeon製品でTensorFlowを動かす方法まとめ。
Installation
基本ソフトウェアをインストール
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にするときも以下の方法が参考になります。
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ドライバのインストール
以前より簡単になっています。
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を指定してインストールします。
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
以上の方法を以下の一行で、実行できます。
curl -sL http://install.aieater.com/setup_rocm_tensorflow_p35 | bash -
References
- TensorFlow-ROCm https://github.com/ROCmSoftwarePlatform/tensorflow-upstream
- ROCm https://github.com/ROCmSoftwarePlatform
- MIOpen https://gpuopen.com/compute-product/miopen/
- GPUEater summarized TensorFlow-ROCm https://github.com/aieater/rocm_tensorflow_info
エンジニア募集中
GPU EATERの開発を一緒に行うメンバーを募集しています。
特にディープラーニング研究者、バックエンドエンジニアを積極採用中です。
募集職種はこちら