深度学习环境tensorflow和GPU(cuda、cudnn)库对应关系2021.11

深度学习环境tensorflow和GPU(cuda、cudnn)库对应关系2021.11

目录

1.tensorflow版本和GPU对应关系

1.1 X86

1.2 macOS

2.tensorflow1和tensorflow2兼容性问题

2.1 改造方法

2.2 测试代码

---

1.tensorflow版本和GPU对应关系 1.1 X86 版本Python 版本cuDNNCUDAtensorflow-2.6.03.6-3.98.111.2tensorflow-2.5.03.6-3.98.111.2tensorflow-2.4.03.6-3.88.011.0tensorflow-2.3.03.5-3.87.610.1tensorflow-2.2.03.5-3.87.610.1tensorflow-2.1.02.7、3.5-3.77.610.1tensorflow-2.0.02.7、3.3-3.77.410.0tensorflow_gpu-1.15.02.7、3.3-3.77.410.0tensorflow_gpu-1.14.02.7、3.3-3.77.410.0tensorflow_gpu-1.13.12.7、3.3-3.77.410.0tensorflow_gpu-1.12.02.7、3.3-3.679tensorflow_gpu-1.11.02.7、3.3-3.679tensorflow_gpu-1.10.02.7、3.3-3.679tensorflow_gpu-1.9.02.7、3.3-3.679tensorflow_gpu-1.8.02.7、3.3-3.679tensorflow_gpu-1.7.02.7、3.3-3.679tensorflow_gpu-1.6.02.7、3.3-3.679tensorflow_gpu-1.5.02.7、3.3-3.679tensorflow_gpu-1.4.02.7、3.3-3.668tensorflow_gpu-1.3.02.7、3.3-3.668tensorflow_gpu-1.2.02.7、3.3-3.65.18tensorflow_gpu-1.1.02.7、3.3-3.65.18tensorflow_gpu-1.0.02.7、3.3-3.65.18 1.2 macOS 版本Python 版本cuDNNCUDAtensorflow_gpu-1.1.02.7、3.3-3.65.18tensorflow_gpu-1.0.02.7、3.3-3.65.18

2.tensorflow1和tensorflow2兼容性问题 2.1 改造方法

如果需要运行一个开源代码，对方环境是tf1，而电脑上安装了tf2，只需要用下面方法改造tf1代码即可：

将tensorflow1代码中的

import tensorflow as tf

替换为

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

由于tensorflow1中

2.2 测试代码

tensorflow1兼容

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
sess = tf.Session()
a = tf.constant(10)
b= tf.constant(12)
print(sess.run(a+b))

tensorflow2

import tensorflow as tf

A = tf.constant([[1, 2], [3, 4]])
B = tf.constant([[5, 6], [7, 8]])
C = tf.matmul(A, B)

print(C)