我正在尝试在Tesla V100-SXM2 GPU上运行CuDNNLSTM层，但是由于安装了TensorFlow-gpu 2.0.0（由于是共享服务器而无法降级），因此会出现错误。

ConfigProto选项在tf 2.0.0时已弃用，因此以前的线程（例如this）无济于事。

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "2"  # Or 2,3,etc. other than 0

tf.config.gpu.set_per_process_memory_growth(True)
tf.config.set_soft_device_placement(True)

如果我使用此代码行，则会显示另一个错误：

  

module notfoundError：没有名为'tensorflow.contrib'的模块