After building TensorFlow from source, seeing libcudart.so and libcudnn errors

后端 未结 12 2559
萌比男神i
萌比男神i 2020-11-28 06:19

I\'m building TensorFlow from source code. The build appears to succeed; however, when my TensorFlow program invokes import tensorflow, one or both of the foll

12条回答
  •  误落风尘
    2020-11-28 06:39

    Mysteriously, my libcudnn.so.5 was installed at ~/cuda/lib64. For people like me, you need to change

    export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:~/cuda/lib64"
    

    to

    export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/yourusername/cuda/lib64"
    

提交回复
热议问题