はじめに
WSLで tensorflow 2.0.0 を動かしたときのメモ。
2020/01/20 最新版に追記修正
環境
ソフトウェア | バージョン | 備考 |
---|---|---|
OS | Windows 10 Pro 64bit | 最新update適用済 |
Ubuntu on WSL | ubuntu 18.04.3 LTS | |
python | python 3.6.9 | pyenv, venv利用 |
tensorflow | 2.1.0 |
前提
- WSL で ubuntu 18.04LTS が動作していること
- 基本的な Linux / python の知識があること
インストール
ubuntu on wsl
ログイン後パッケージをアップデート。必要なパッケージをインストール
$ sudo apt update
$ sudo apt upgrade
$ sudo apt install bzip2 zlib1g-dev libssl-dev readline-common sqlite3
$ sudo apt install libssl-dev libbz2-dev libreadline-dev libsqlite3-dev zlib1g-dev libffi-dev gcc make
sqlite3などは念のため。
pyenv
Githubの pyenv-installer を使う方法でインストール
$ curl https://pyenv.run | bash
$ curl -L https://github.com/pyenv/pyenv-installer/raw/master/bin/pyenv-installer | bash
~/.bashrc に以下を追加し、source ~/.bashrc などで反映。 username は実際のユーザ名にする
export PATH="/home/username/.pyenv/bin:$PATH"
eval "$(pyenv init -)"
eval "$(pyenv virtualenv-init -)"
Bash on Windows 起動時に~/.bashrcを参照するように ~/.bash_profile に以下の内容を追加。
if [[ -f ~/.bashrc ]] ; then
. ~/.bashrc
fi
python
準備
sudo apt install gcc bzip2 zlib1g-dev libssl-dev readline-common libnss3-tools
コンパイルしてインストール
$ pyenv install -list # インストールできるバージョンの確認
$ pyenv install 3.6.9
$ pyenv versions # インストールバージョンの確認
$ pyenv global 3.6.9 # 全体のバージョン指定、必要に応じて変更
$ python --version # python のバージョン確認
python 3.7.5 だと Tensolflow のインストールに失敗したので、3.6系の最新である 3.6.9 を利用。原因不明。
venv仮想環境設定
$ cd work # 任意のディレクトリに移動
$ python -m venv tf2
$ ls -la ./tf2/bin # ディレクトリの存在確認
$ source ./tf2/bin/activate # 有効化、常時使う場合は.bashrcなどに書いておく
TensolFlow本体インストール
GPUを使わない場合
$ pip install --upgrade pip # pipの更新
$ pip install tensorflow
以上でインストール完了。tensorflowのアッグレードは次のように行う。
$ pip install --upgrade tensorflow
関連ライブラリ等をインストール
必要に応じて開発環境、Pythonライブラリ群をインストール
まずpipでpyhonライブラリをインストール
$ pip install tensorflow_datasets
$ pip install numpy pandas pillow opencv-python
(参考)CUDA経由でGPUを使う場合
以下のような手順が必要らしい(未確認)
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 dpkg -i 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
sudo apt update
apt upgrade で、GPG error: ..... NO_PUBKEY .....が出た場合は下記コマンドを実行。
curl -s -L http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key add -
参考
- https://www.tensorflow.org/install
- https://www.tensorflow.org/tutorials
- https://qiita.com/nuka137/items/6966efeddceb96012819
- https://qiita.com/shoji9x9/items/5110ef7f59c678d07aa0
- https://qiita.com/imsk/items/d14be2a7fcca8a080a8d
蛇足
昔、インストールでハマったのがウソみたい。
でもバージョンアップでハマった。開発環境あるある。