Which TensorFlow and CUDA version combinations are compatible?

后端 未结 6 685
慢半拍i
慢半拍i 2020-11-22 13:07

I have noticed that some newer TensorFlow versions are incompatible with older CUDA and cuDNN versions. Does an overview of the compatible versions or even a list of officia

6条回答
  •  粉色の甜心
    2020-11-22 13:21

    The compatibility table given in the tensorflow site does not contain specific minor versions for cuda and cuDNN. However, if the specific versions are not met, there will be an error when you try to use tensorflow.

    For tensorflow-gpu==1.12.0 and cuda==9.0, the compatible cuDNN version is 7.1.4, which can be downloaded from here after registration.

    You can check your cuda version using
    nvcc --version

    cuDNN version using
    cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

    tensorflow-gpu version using
    pip freeze | grep tensorflow-gpu

    UPDATE: Since tensorflow 2.0, has been released, I will share the compatible cuda and cuDNN versions for it as well (for Ubuntu 18.04).

    • tensorflow-gpu = 2.0.0
    • cuda = 10.0
    • cuDNN = 7.6.0

提交回复
热议问题