Ubuntu20 + DockerでStylegan2を動かす

ドライバ、Toolkitのダウンロード

CUDA Toolkitを以下からダウンロードしておく。
https://developer.nvidia.com/cuda-downloads

CUDA Toolkitのダウンロード方法は複数あるが、おすすめはruntime (local)オプション

最新のドライバーをインストールするため、GUIではなくCUIで起動している必要がある。
以下コマンドを実行

sudo systemctl set-default multi-user.target

その後再起動することで、UbuntuがCUIで起動するようになる。

しかし筆者の環境では再起動後に真っ黒な画面のまま進まなくなってしまった。
https://askubuntu.com/questions/1099463/cannot-boot-after-sudo-systemctl-set-default-multi-user-target
を参考に「Ctrl + alt + f2」を同時押しすることで解決できた。

set-defaultではなくisolateを使えば一時的に変更できるみたい。

先程ダウンロードしたパッケージをインストールする。

最初にドライバをインストールするため、ダウンロードしたフォルダ内で以下を実行

sudo sh NVIDIA-Linux-x86_64-470.82.00.run

ウェザートにしたがって完了後、同様にcuda toolkitをインストール

sudo sh cuda_11.5.0_495.29.05_linux.run

https://github.com/NVlabs/stylegan2
のRequirementsに書いてあるように、リポジトリをクローンしたフォルダの中で、

nvcc test_nvcc.cu -o test_nvcc -run

を実行してみる。正しくできていれば

| CPU says hello. | GPU says hello.

と出力される。

以下コマンドを実行後、再起動

sudo systemctl set-default graphical.target

このページを参考にインストールする。

Dockerfileが同梱されているので、それを利用する。
https://github.com/NVlabs/stylegan2/blob/master/Dockerfile

ビルドする。

docker build . -t stylegan2

起動。
--gpus allオプションを忘れずにつける。

sudo docker run -it --gpus all stylegan2

ビルドしたとはいうもののまっさらの状態なので、gitを入れリポジトリをクローンしてくる。

# apt update && apt install -y git && git clone https://github.com/NVlabs/stylegan2

Stylegan2、READMEの「Using pre-trained networks」にあるコマンドを実行してみる

# python run_generator.py generate-images --network=gdrive:networks/stylegan2-ffhq-config-f.pkl \
  --seeds=6600-6625 --truncation-psi=0.5

同resultsフォルダに生成された画像が保存される。