tensorflow 安装

tensorflow清华镜像




tensorflow1.4.0 支持CUDA 8 和 Cudnn v6
安装CUDNN

tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

建立软链接

cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.6
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -s libcudnn.so.6 libcudnn.so

设置环境变量 sudo vim /etc/profile 添加如下内容

PATH=/usr/local/cuda/bin:$PATH
export PATH

保存后,创建链接文件
sudo vim /etc/ld.so.conf.d/cuda.conf
添加如下内容
/usr/local/cuda/lib
最后使链接生效
sudo ldconfig

Python2.7 安装Tensorflow1.4.0 -gpu

pip install tensorflow_gpu-1.4.0-cp27-none-linux_x86_64.whl