version error of tensorflow & gpu

[ChengYaofeng]

It's a nice idea and appreciate for opening source.
When I replicated this code, I met the question that the 1.15 Tensorflow version is not satisfied because my 3090GPU can only work with cuda11.
Then I used Tensorflow 2.12, and added the tf.disable_v2_behavior, but the Traceback said that from tensorflow.contrib.model_pruning.python import pruning.
So, I wonder what the version of your virtual environment and the gpu you use, can you tell me?