StarGAN-Keras icon indicating copy to clipboard operation
StarGAN-Keras copied to clipboard

Updated code for Tensorflow 2.5 and CUDA 11.3

Open Atif-Anwer opened this issue 3 years ago • 0 comments

Updated requirements:

The code was updated and last tested scuessfully with the following updated packages:

Tensorflow: 2.5.0
Keras: 2.4.3
OpenCV: 4.5.1
Numpy: 1.20.3 (also tried 1.19.2)
Python: 3.8.5
OS: Ubuntu 20.04 LTS

The training works on the following nvidia drivers:

NVIDIA Driver Version: 465.19.01
CUDA Version: 11.3
cuDNN Version: 8.1.1

NOTE: You can skip merging the readme.md since i changed it for myself.

Atif-Anwer avatar Jun 10 '21 09:06 Atif-Anwer