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
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。