Open zizhuxuaner opened 3 years ago
Hi, this project was built with TensorFlow 1.x.x which contains the module tensorflow.contrib. However, tensorflow.contrib was removed in TensorFlow 2.x.x.
Can you try to install the version mentioned in the README and see if it works ?
tensorflow-gpu 1.8.0 (ubuntu, GPU 1080Ti or Titan xp, cuda 9.0, cudnn 7.1.3)
tensorflow-gpu 1.15.0 (ubuntu, GPU 2080Ti, cuda 10.0.130, cudnn 7.6.0)
ModuleNotFoundError: No module named 'tensorflow.contrib'