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

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

    First, Install CUDA library (version 7.5) from here

    Installation Instructions: 1- sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb 2- sudo apt-get update 3- sudo apt-get install cuda

    Second, install the cuDNN from here

    Third, export cuDNN path:

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

    In case you have an error like "The package libcudnnX needs to be reinstalled", follow those steps here

提交回复
热议问题