安装tensorflow
下载CUDNN
十分的不和谐,居然强制要求注册登陆,差评。
https://developer.nvidia.com/rdp/cudnn-download
登陆后选择需要的版本下载即可
添加CUDNN到cuda-8.0库
安装Tensorflow
下载whl文件
https://github.com/tensorflow/tensorflow
最下面README中有选择版本,下载即可。
然后执行
运行tensorflow实例程序
一个挺不错的中文文档:
http://wiki.jikexueyuan.com/project/tensorflow-zh/get_started/os_setup.html
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*
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