Support for Nvidia CUDA Toolkit 9.2

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

    The issue is not with NVIDIA drivers but Tensorflow itself. I spent an hour trying to make it work, and finally realized that if you download the pre-built binary from googleapi.com, it is hard coded to load libcudart.so.9.0! If you have both cuda 9.0 and 9.2 installed, tensorflow will work (but it's actually loading the dynamic libraries from 9.0). (BTW, I installed TF using anaconda.)

    A cleaner approach is to build TF from source. It's not too complicated.

提交回复
热议问题