首先,下载需要安装的包 cuda(8.0),cudnn(6.0),tensorflow-1.3
将所有文件放入主目录下的一个文件夹,我放在~/fk/111
然后执行下面这个.sh 文件就可以了
.sh文件的内容如下:
#!/bin/bash
#sudo tar xzf ~/fk.tar.gz
cd ~/fk/111/
sudo apt-get install -y gcc g++ gfortran build-essential git wget linux-image-generic libopenblas-dev python-dev python-pip python-nose python-numpy python-scipy Python-Opencv
sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo chmod +777 -R /opt/
sudo chmod +777 -R /data/
sudo mkdir ~/.pip/
echo "[global]
timeout = 60
index-url = http://pypi.douban.com/simple
trusted-host = pypi.douban.com" >>~/.pip/pip.conf
sudo apt-get install cuda
#sudo cp -a profile /etc/profile
source /etc/profile
tar xzf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo pip install -U pip
sudo pip install ~/fk/111/tensorflow_gpu-1.3.0-cp27-cp27mu-manylinux1_x86_64.whl
source /etc/profile
#sudo rm -fr cuda
#sudo tar xf cudnn-7.5-linux-x64-v5.1.tar
#sudo cp cuda/include/cudnn.h /usr/local/cuda/include
#sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64
由于我是批量安装,所有其中的profile文件是我在原来的文件后面加了环境变量
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
你可以在开始先对profile进行操作,然后省却sudo cp -a profile /etc/profile 这条命令
后面三行是我又将cudnn6.0换成5.0,不会影像tensorflow的使用