はじめに
先日PCのGPUをGTX1080からRTX3080に載せ換えました。RTX3000番台のGPUはCUDA 10.xをサポートしていないので古いTensorFlowは動かせないものと思っていました。調べてみるとそうでもないようです。NVIDIAがCUDA 11.xで動かせるTensorFlowを公開してくれています。さっそく試してみました。3年前に試したSC-FEGANをRTX3080で動かしていきます。touch-sp.hatenablog.com
SC-FEGANの公式リポジトリはこちらです。
github.com
実行環境
Ubuntu 20.04 on WSL2 CUDA 11.3.1 Python 3.8.10
PyQt5を使うのでその設定が必要です。こちらを参照して下さい。
Python環境構築
パッケージのインストールはすべてpipを用いて行いました。nvidia-tensorflowもpipで簡単にインストール可能です。nvidia-tensorflowとインストールしているCUDAのバージョンをあわすためにこちらを参照して下さい。pip install nvidia-pyindex pip install nvidia-tensorflow==1.15.5+nv21.06 pip uninstall tensorboard pip install nvidia-tensorboard pip uninstall protobuf pip install protobuf==3.20.0 pip install pyqt5==5.12.2 pip install pyyaml==5.1.1 pip install opencv-python pip install mock pip install pbr
absl-py==1.2.0 astor==0.8.1 astunparse==1.6.3 gast==0.3.3 google-pasta==0.2.0 grpcio==1.47.0 h5py==2.10.0 importlib-metadata==4.12.0 Keras-Applications==1.0.8 Keras-Preprocessing==1.1.2 Markdown==3.4.1 MarkupSafe==2.1.1 mock==4.0.3 numpy==1.18.5 nvidia-cublas==11.5.1.101 nvidia-cublas-cu113==11.5.1.109 nvidia-cuda-cupti-cu113==11.3.111 nvidia-cuda-nvcc-cu113==11.3.109 nvidia-cuda-nvrtc==11.1.105 nvidia-cuda-runtime==11.3.58 nvidia-cuda-runtime-cu113==11.3.109 nvidia-cudnn==8.2.0.51 nvidia-cudnn-cu113==8.2.1.32 nvidia-cufft-cu113==10.4.2.109 nvidia-curand-cu113==10.2.4.109 nvidia-cusolver-cu113==11.1.2.109 nvidia-cusparse-cu113==11.6.0.109 nvidia-dali-cuda110==1.2.0 nvidia-dali-nvtf-plugin==1.2.0+nv21.6 nvidia-nccl-cu113==2.9.9 nvidia-pyindex==1.0.9 nvidia-tensorboard==1.15.0+nv21.4 nvidia-tensorflow==1.15.5+nv21.6 nvidia-tensorrt==7.2.3.4 opencv-python==4.6.0.66 opt-einsum==3.3.0 pbr==5.9.0 pkg_resources==0.0.0 protobuf==3.20.0 PyQt5==5.12.2 PyQt5-sip==12.11.0 PyYAML==5.1.1 six==1.16.0 tensorboard @ https://pypi.ngc.nvidia.com/tensorboard/tensorboard-1.15.0-py2.py3-none-any.whl tensorflow-estimator==1.15.1 termcolor==1.1.0 webencodings==0.5.1 Werkzeug==2.2.1 wrapt==1.14.1 zipp==3.8.1
実行
まずはGitHubをクローンして「ckpt」フォルダを作成します。git clone https://github.com/run-youngjoo/SC-FEGAN.git cd SC-FEGAN mkdir ckpt
「SC-FEGAN.ckpt.data-00000-of-00001」「SC-FEGAN.ckpt.index」の二つをこちらからダウンロードして「ckpt」フォルダに保存します。
最後に以下を実行するだけです。
python demo.py
補足
今回はフリー素材「ぱくたそ」から顔写真を使わせて頂きました。こちらの写真です。
