Tips/Ubuntu/機械学習環境設定 のバックアップ(No.1)



前提条件

Ubuntu 18.04.3 LTS (Bionic Beaver)

nvidia driver

  1. 標準ドライバを削除
    sudo apt --purge nvidia*
  2. 最新ドライバをインストールし再起動
    sudo ubuntu-drivers autoinstall
    reboot
  3. 確認
    nvidia-smi

機械学習環境設定

  1. 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
  2. 機械学習リポジトリ追加
    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
  3. CUDA&cuDNNインストール
    sudo apt install --no-install-recommends cuda-10-0 libcudnn7 libcudnn7-dev -y
  4. TensorRTインストール
    sudo apt install --no-install-recommends libnvinfer5 libnvinfer-dev -y
  5. 関連ライブラリインストール
    sudo apt install build-essential pkg-config libopenblas-dev liblapack-dev libhdf5-serial-dev graphviz -y
  6. python3インストール
    sudo apt install python3-venv python3-pip python3-dev python3-tk -y

jupyter notebook設定

  1. nbextensionsインストール
    pip install jupyter-contrib-nbextensions
    pip install jupyter-nbextensions-configurator
  2. nbextensions有効化
    jupyter contrib nbextension install --user
    jupyter nbextensions_configurator enable --user
  3. jupyter-notebook 起動
    jupyter-notebook --ip=*cc