Tips/Ubuntu/機械学習環境設定 の変更点

Top > Tips > Ubuntu > 機械学習環境設定

#contents
----
* 目的 [#vb066fcd]
- nvidia driverをインストールする
- NVIDIAが提供するリポジトリを利用して機械学習関連の環境を設定する
- jyupyter notebookを設定する
* 前提条件 [#a3fa0bc2]
Ubuntu 18.04.3 LTS (Bionic Beaver)

* 自動アップデートを禁止する [#z80e44eb]
- アップデート設定をOFFする
カーネルが自動更新されるとnvidiaドライバが動作しなくなったりする
/etc/apt/apt.conf.d/20auto-upgrades

 APT::Periodic::Unattended-Upgrade "1";
#geshi(bash,number){{
APT::Periodic::Unattended-Upgrade "1";
}}

 APT::Periodic::Unattended-Upgrade "0";
#geshi(bash,number){{
APT::Periodic::Unattended-Upgrade "0";
}}
* nvidia driver [#ye2af00f]
+ 標準ドライバを削除
 sudo apt --purge remove nvidia-*
#geshi(bash,number){{
sudo apt --purge remove nvidia-*
}}
+ 最新ドライバをインストールし再起動
 sudo ubuntu-drivers autoinstall
 sudo reboot
#geshi(bash,number){{
sudo ubuntu-drivers autoinstall
sudo reboot
}}
+確認
 nvidia-smi
#geshi(bash,number){{
nvidia-smi
}}
-参考
--[[UbuntuにNvidia GPUのDriver + CUDAをInstallする(GTX1080対応版)>https://qiita.com/conta_/items/d639ef0068c9b7a0cd12]]
* 機械学習環境設定[#q11100cd]
+ cudaリポジトリ追加
 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
 sudo apt install ./cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
#geshi(bash,number){{
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt install ./cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
}}
+ 公開鍵を追加する
cudaリポジトリ追加時に表示されるコマンドで公開鍵を追加する
 sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
#geshi(bash,number){{
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
}}
+ 機械学習リポジトリ追加
 wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
 sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
#geshi(bash,number){{
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
}}
+ CUDA&cuDNNインストール
 sudo apt update
 sudo apt install --no-install-recommends cuda-10-0 libcudnn7 libcudnn7-dev -y
#geshi(bash,number){{
sudo apt update
sudo apt install --no-install-recommends cuda-10-0 libcudnn7 libcudnn7-dev -y
}}
+ TensorRTインストール
 sudo apt install --no-install-recommends libnvinfer5 libnvinfer-dev -y
#geshi(bash,number){{
sudo apt install --no-install-recommends libnvinfer5 libnvinfer-dev -y
}}
+ 関連ライブラリインストール
 sudo apt install build-essential pkg-config libopenblas-dev liblapack-dev libhdf5-serial-dev graphviz -y
#geshi(bash,number){{
sudo apt install build-essential pkg-config libopenblas-dev liblapack-dev libhdf5-serial-dev graphviz -y
}}
+ python3インストール
 sudo apt install python3-venv python3-pip python3-dev python3-tk -y
#geshi(bash,number){{
sudo apt install python3-venv python3-pip python3-dev python3-tk -y
}}
+ 各種ライブラリをインストール
 sudo -H pip3 install numpy scipy matplotlib pyyaml h5py\
  pydot-ng opencv-python keras jupyter pillow python-dateutil
 sudo -H pip3 install tensorflow-gpu

#geshi(bash,number){{
sudo -H pip3 install numpy scipy matplotlib pyyaml h5py\
 pydot-ng opencv-python keras jupyter pillow python-dateutil
sudo -H pip3 install tensorflow-gpu
}}
-参考
-- [[Ubuntu 18.04 で、NVIDIAのリポジトリを利用してtensorflow-gpu環境を構築するシンプルな方法>https://www.nemotos.net/?p=3176]]
*jupyter notebook設定 [#dc87ba0a]
jyupyter notebookの拡張機能をインストールする
+ nbextensionsインストール
 sudo -H pip3 install jupyter-contrib-nbextensions
 sudo -H pip3 install jupyter-nbextensions-configurator
#geshi(bash,number){{
sudo -H pip3 install jupyter-contrib-nbextensions
sudo -H pip3 install jupyter-nbextensions-configurator
}}
+ nbextensions有効化
 jupyter contrib nbextension install --user
 jupyter nbextensions_configurator enable --user
#geshi(bash,number){{
jupyter contrib nbextension install --user
jupyter nbextensions_configurator enable --user
}}
+ jupyter-notebook 起動
 jupyter-notebook --ip=* &
#geshi(bash,number){{
jupyter-notebook --ip=* &
}}
-参考
--[[Jupyter Notebook - おすすめの nbextensions まとめ>https://www.pynote.info/entry/jupyter-notebook-nbextensions]]