Which TensorFlow and CUDA version combinations are compatible?

后端 未结 6 802
慢半拍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:26

    TL;DR) See this table: https://www.tensorflow.org/install/source#gpu

    Generally:

    Check the CUDA version:

    cat /usr/local/cuda/version.txt
    

    and cuDNN version:

    grep CUDNN_MAJOR -A 2 /usr/local/cuda/include/cudnn.h
    

    and install a combination as given below in the images or here.

    The following images and the link provide an overview of the officially supported/tested combinations of CUDA and TensorFlow on Linux, macOS and Windows:

    Minor configurations:

    Since the given specifications below in some cases might be too broad, here is one specific configuration that works:

    • tensorflow-gpu==1.12.0
    • cuda==9.0
    • cuDNN==7.1.4

    The corresponding cudnn can be downloaded here.

    Tested build configurations

    Please refer to https://www.tensorflow.org/install/source#gpu for a up-to-date compatibility chart (for official TF wheels).

    (figures updated May 20, 2020)

    Linux GPU

    Linux

    macOS GPU

    macOS

    (figure updated May 31, 2018)

    Windows

    Updated as of 14th Jan 2020: For the updated information please refer Link for Linux and Link for Windows.

提交回复
热议问题