Support for Nvidia CUDA Toolkit 9.2

后端 未结 3 1043
情深已故
情深已故 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:31

    That's a good question. According to NVidia's website,

    The CUDA driver is backward compatible, meaning that applications compiled against a particular version of the CUDA will continue to work on subsequent (later) driver releases.

    So technically, it should not be a problem to support later iterations of a CUDA driver. And in practice, you will find working non-official pre-built binaries with later versions of CUDA and CuDNN on the net [1], [2]. Even easier to install, the tensorflow-gpu package installed from conda currently comes bundled with CUDA 9.2.

    When asked on the topic, a dev answered,

    The answer to why is driver issues in the ones required by 9.1, not many new features we need in cuda 9.1, and a few more minor issues.

    So the reason looks rather vague -- he might mean that CUDA 9.1 (and 9.2) requires graphics card driver that are perhaps a bit too recent to be really convenient, but that is an uneducated guess.

    If NVidia is right about binary compatibility, you may try to simply rename or link your CUDA 9.2 library as a CUDA 9.0 library and it should work. But I would save all my work before attempting this... and the fact that people go as far as recompiling tensorflow to support later CUDA versions may be a hint on how this could end.

提交回复
热议问题