显卡驱动是安装tensorflow最操蛋的,也是最难受的,深受其害,耽误半天时间。
按装这个驱动之前首先要确定你电脑的显卡是英伟达的,否则累死你也做不上,还骂了半天的娘。
网上最常用的方法如下:
在ubuntu16.04中,更换驱动非常方便,去
系统设置->软件更新->附加驱动->切换到最新的NVIDIA驱动即可。应用更改->重启
然后输入
nvidia-smi
命令
如果没有这个框的话很有可能,就是
nvida
的版本太低,你要重新下载一个新版本,越新越好。亲测有效,解决了大毛病。
更新
nvidia
方法:
sudoapt purge nvidia-*
sudoadd-apt-repository ppa:graphics-drivers/ppa
sudoapt update
sudoapt install nvidia-387
二、安装
JDK8
sudoapt-get install software-properties-common
sudoadd-apt-repository ppa:webupd8team/java
sudoapt-get update
sudoapt-get install oracle-java8-installer
验证
java
版本:
java-version
三、安装
CUDA
9.0
安装cuda时要清楚你的软件支持哪一版的。
下载地址:
https://developer.nvidia.com/cuda-downloads
四、安装
CUDNN
7.1.4
下载地址:
https://developer.nvidia.com/cudnn
tar -xvf cudnn-
9
.0-linux-x64-v
7
.
1
.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*
五、配置环境变量
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
exportPATH="$CUDA_HOME/bin:$PATH"
source ~/.bashrc
六、安装TensorFlow
1、安装依赖工具
sudoapt-get install python-numpy swig python-dev python-wheel
2
、用
pip
安装很方便
pipinstall tensorflow_gpu==1.8
问题:
1、没有libcudnn.so.7cd
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.1.4 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so