ubuntu16.04下tensorflow-0.10(cuda7.5)安装

来源:互联网 发布:长沙蓝狐网络官网 编辑:程序博客网 时间:2024/04/29 16:12

tensorflow-0.10(cuda7.5)安装


1.annaconda(python2.7)、cuda(7.5)、cudnn(5.1)参见另一篇文章《

ubuntu--pva-faster-rcnn》点击打开链接




2.tensorflow-0.10安装


     Ubuntu/Linux 64-bit, CPU only, Python 2.7:

    $sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl  


    Ubuntu/Linux 64-bit, GPU enabled, Python 2.7. Requires CUDA toolkit 7.5 and CuDNN v5.1:

   $sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl




问题来啦:

1.could not find cudnnCreate in cudnn DSO; dlerror: /usr/local/lib/python2.7/dist-packages/tensorflow/python/_pywrap_tensorflow.so: undefined symbol: cudnnCreate

解决办法

export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"

sudo ldconfig /usr/local/cuda/lib64


2.libjpeg.so.62: cannot open shared object file: No such file or directory

solution: sudo apt-get install libjpeg62-dev

0 1
原创粉丝点击