
GitHub - NVlabs/stylegan2: StyleGAN2 - Official TensorFlow Implementation
StyleGAN2 - Official TensorFlow Implementation. Contribute to NVlabs/stylegan2 development by creating an account on GitHub.
上記リポジトリからcloneしたStyleGAN2を実行した時にエラーが発生。
int() argument must be a string, a bytes-like object or a number, not 'Tensor'
エラーログを追いかけてみると、cudnnのバージョンが違うよと怒られていた。
Loaded runtime CuDNN library: 7.4.2 but source was compiled with: 7.6.0. CuDNN library major and minor version needs to match or have higher m
inor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library.
cudnnをダウンロードし、古いバージョンをアンインストールしてから7.6.0にアップグレード。
# どちらか一方を実行
sudo dpkg -i libcudnn*.deb
aptitude remove libcudnn*
sudo dpkg -i libcudnn7_7.6.0-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.0.64-1+cuda10.0_amd64
sudo dpkg -i libcudnn7-doc_7.6.0.64-1+cuda10.0_amd64
これでStyleGAN2のトレーニングを始めることが出来た。
参考
Installation Guide - NVIDIA Docs
This cuDNN 8.9.5 Installation Guide provides step-by-step instructions on how to install and check for correct operation of NVIDIA cuDNN on Linux and Microsoft ...
Installation Guide - NVIDIA Docs
This cuDNN 8.9.5 Installation Guide provides step-by-step instructions on how to install and check for correct operation of NVIDIA cuDNN on Linux and Microsoft ...
コメント