【備忘録】Ubuntu18.04のCUDA10&Python3&PyTorch環境構築

研究室のPCの環境構築を頼まれたので，また構築を頼まれた時に楽ができるようにメモしておきます．

環境

• OS : Ubuntu 18.04.1 LTS 64bit
• CPU : Intel core i3 @ 4.00GHz x 4
• GPU : GeForce GTX 1080 Ti

CUDA 10.0

CUDA Toolkit 10.0 Download | NVIDIA Developer
Installer Type で deb(network) を選択する場合

$ sudo apt update
$ sudo apt upgrade

$ 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/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
$ sudo apt update

$ sudo apt install cuda cuda-drivers  # cuda-10-0でver指定も可能
$ sudo reboot  # 再起動

$ rm cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

$ export PATH="/usr/local/cuda/bin:$PATH"
$ export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"

$ exit  # 端末を起動しなおす

$ nvidia-smi  # 確認

$ sudo apt autoremove
$ sudo apt clean

間違えてUbuntu16.04向けのものをダウンロードしたらaptの依存関係エラーで大変なことになった...

Anaconda

下記サイトからインストーラーをダウンロード
Downloads - Anaconda

2系を使う予定がなかったので Python 3.7 version を選択（2系なら2.7）

$ bash ./Anaconda3-5.3.1-Linux-x86_64.sh  # .shファイルが保存されているディレクトリで実行

ラインセンスを読んで同意し，インストール場所を設定（デフォルトはEnter）するとインストールが開始

$ conda -V  # 確認
conda 4.5.11

仮想環境構築

環境名とPythonのバージョン，インストールしたいパッケージを羅列して実行

$ conda create -n 環境名 Python=x.x パッケージ

例）

$ conda create -n py37 python=3.7 anaconda
Proceed ([y]|n)? y

アクティブ

$ conda activate 環境名

確認

$ conda info -e
# conda environments:
#
base               * /home/.../anaconda3
py37                 /home/.../anaconda3/envs/py37

$ conda activate 環境名
$ python -V
Python 3.7.1

非アクティブ

$ conda deactivate

仮想環境の削除

$ conda remove -n 環境名 --all

PyTorch

PyTorch 公式HP
Get Start から自分の環境を指定するとインストールコマンドが出力される．

$ conda install pytorch torchvision cuda100 -c pytorch

GPU動作確認

その１

$ python
>>> import torch
>>> nums = torch.randn(2,2)
>>> nums.cuda()  # 結果が出ればおｋ

その2

$ git clone https://github.com/pytorch/examples.git
$ cd examples/vae/
$ python main.py  # 結果がでればおｋ

A repository showcasing examples of using PyTorch

実行中に $ nvidia-smi -l を実行して確認する．

メモ

VisualStudioCodeのインストール

Visual Studio Code - Code Editing. Redefined

Gitのインストール

$ sudo apt-get install git

参考にしました

Ubuntu 18.04へのCUDAインストール方法 - Qiita
Ubuntuを初めて触るド素人がAnacondaを入れてみた - Qiita
Markdown記法 チートシート - Qiita
WindowsでCUDA+PyTorchのサンプル実行するまでのメモ - Qiita