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

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

    As of now tensor-flow supports cuda-9.0

    Do following things. Hope it helps :

    $ sudo apt-get install cuda-9.0
    $ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
    

    Download cuDNN for 9.0 (You need to register before downloading) https://developer.nvidia.com/rdp/form/cudnn-download-survey

    $ sudo dpkg -i libcudnn7_7.1.2.21-1+cuda9.0_amd64.deb
    

    Close all terminal and open new

    $ source activate tensorflow
    $ python
    >> import tensorflow as tf
    

    You should not get any error after this.

提交回复
热议问题