Support for Nvidia CUDA Toolkit 9.2

后端 未结 3 1051
情深已故
情深已故 2021-02-09 08:07

What is the reasoning that Tensorflow-gpu is bound to a specific version of Nvidia\'s CUDA Toolkit? The current version appears to look for 9.0 specifically and will not work wi

3条回答
  •  旧巷少年郎
    2021-02-09 08:39

    When you download TF, you download a pre-built binary file. In the build process TF is hard linked into a specific version of Cuda, so you cannot use it with different cuda versions.

    If you want to work with the new (or sometimes older) version of cuda you will need to install TF from source (check how here) Or, if you realy don't want to build yourself, check in these repos, there are others that publish specific TF binaries, few examples:

    • https://github.com/mind/wheels
    • https://github.com/yaroslavvb/tensorflow-community-wheels
    • https://github.com/fo40225/tensorflow-windows-wheel

    For your convenience I add here the CUDA + cuDNN versions that are required for each prebuilt Tensorflow version:

    (I write here just about the TF versions that I worked with, maybe older TF versions use older versions of CUDA as well)

    • before TF v1.5 cuda 8.0 and cuDNN 6
    • start from: 1.5 - Prebuilt binaries are now built against CUDA 9 and cuDNN 7.

提交回复
热议问题