StyleGAN2でエラーint() argument must be a string, a bytes-like object or a number, not ‘Tensor’

GitHub - NVlabs/stylegan2: StyleGAN2 - Official TensorFlow Implementation

StyleGAN2 - Official TensorFlow Implementation. Contribute to NVlabs/stylegan2 development by creating an account on Git...

上記リポジトリから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のトレーニングを始めることが出来た。

参考

https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installlinux

https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#upgrade