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

后端 未结 12 2558
萌比男神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:32

    I fixed this just adding the cuda path to my .bashrc

    export LD_LIBRARY_PATH=/usr/local/cuda/lib64/
    

    Just have in mind that first you need to go to nvidia Deep Learning page, register and download cuDNN, extract and copy the files from include and lib64 folders into your cuda installation.

提交回复
热议问题