FunnyWeb

据说这里有很多神奇的东西...

安装tensorflow

2017-04-29 14:32:552112 views
  1. 下载CUDNN

    十分的不和谐,居然强制要求注册登陆,差评。

    https://developer.nvidia.com/rdp/cudnn-download

    登陆后选择需要的版本下载即可

  2. 添加CUDNN到cuda-8.0库

  3. sudo tar xvzf cudnn-8.0-linux-x64-v5.0-ga.tgz  
    sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include  
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64  
    sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
  4. 安装Tensorflow

    下载whl文件

    https://github.com/tensorflow/tensorflow

    最下面README中有选择版本,下载即可。

    然后执行

  5. sudo apt-get install python-pip python-dev
    sudo pip install --ignore-installed --upgrade tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl
  6. 运行tensorflow实例程序

    一个挺不错的中文文档:

    http://wiki.jikexueyuan.com/project/tensorflow-zh/get_started/os_setup.html