这几天在安装TensorFlow,看了很多教程,方法也试了几种。
最后还是用pip安装成功的,过程如下。
1.安装ubuntu后在 系统设置-软件与更新-附加驱动 里,更新N卡驱动。
(N卡官网下载run格式驱动手动安装比较麻烦,需要关闭X server,并禁止系统自带的开源驱动)
按照提示,重启。重启之后,终端输入nvidia-smi,可以看到GPU的信息。
2.N卡开发者官网下载CUDA deb格式安装包,这里注意,CUDA、cudnn、TensorFlow之间版本有要求。我最后使用的是cuda8.0+cudnn6.0+TensorFlow1.4
下载cuda的deb格式安装包,然后安装
#进入下载文件所在目录,执行下列命令:
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
3.N卡开发者官网下载cudnn,需要注册并填写一个问卷。
#进入下载文件所在目录,执行下列命令:
tar xvzf 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/