一、 正常安装Ubuntu16.04,注意英文环境,否则切换到命令界面中文会变成菱形乱码
二、 安装显卡驱动
sudo chmod 666 /etc/modprobe.d/blacklist.conf
sudo gedit /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb
sudo update-initramfs -u
sudo service lightdm stop
sudo /etc/init.d/lightdm stop(后面这个有效,前面这个时灵时不灵)
不用管,继续
选择“no"
sudo apt-get install linux-image-extra-virtual
否则后续可能会出现如下错误提示:
modprobe:ERROR: could not insert'nvidia':Unknown symbolinmodule,or unknown parameter(seedmesg)
三、 安装cuda 8.0
sudo gedit /etc/profile
在末尾处添加(注意不要有空格,不然会报错):
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH
测试cuda的Samples
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如果显示的是一些关于GPU的信息,则说明安装成功了。
四、 安装cuDNN(安装时tensorflow更新到1.3,需用6.0;预计1.4用cuDNN7.0)
libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
libcudnn6_6.0.21-1+cuda8.0_amd64.deb
sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb
sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
注意顺序别反
cuDNN6.0已经把环境变量自动添加进去了,所以我没有再手动添加
五、 在 bashrc 中添加安装位置
sudo gedit ~/.bashrc
exportLD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
六、 安装带有 GPU 支持的 TensorFlow
因为我的python是3.5,所以按下面的命令安装
pip3 installTensorFlow
七、 可以在python里面import tensorflow测试一下