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のトレーニングを始めることが出来た。

参考

Page Not Found

Page displayed for direct links that are not supported by current publishing formats.

Page Not Found

Page displayed for direct links that are not supported by current publishing formats.