作者:冯拓
电脑配置如下:
配置 |
HP-Z820 |
CPU核心线程数和主频 |
intel xeon(至强) E-5 2620 2.0GHz*24 |
内存 |
64GB |
硬盘 |
2TB |
显卡 |
NIVDIA TITAN X 12GB |
安装过程中使用的安装包:
|
安装包 |
驱动 |
NVIDIA-Linux-x86_64-396.18.run |
cuda |
cuda_9.1.85_387.26_linux.run |
cudnn |
cudnn-9.1-linux-x64-v7.1.tgz |
Anaconda |
Anaconda3-5.2.0-Linux-x86_64.sh |
bazel |
bazel-0.14.1-installer-linux-x86_64.sh |
tensorflow源代码 |
tensorflow-r1.8 |
cudnn |
cudnn-9.1-linux-x64-v7.1.tgz |
cuda下载链接:
https://developer.nvidia.com/cuda-downloads
anaconda下载链接:
https://www.continuum.io/downloads
bazel下载链接:
https://github.com/bazelbuild/bazel/releases
tensorflow源代码下载链接:
https://github.com/tensorflow/tensorflow
本文分为两个大部分来介绍,首先介绍cuda与cudnn的安装,然后介绍使用源码安装tensorflow。
一、安装cuda与cudnn
1、Nvidia驱动安装-run文件安装
下载完名称为NVIDIA-Linux-x86_64-396.18.run 的文件。使用以下命令:
sudo gedit /etc/modprobe.d/blacklist.conf
加入以下语句,将nouveau禁止命令写入文件。
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
调用指令禁止nouveau。
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
建立新的内核并重启。
sudo update-initramfs -u
sudo reboot
使用ctrl + alt+ F1进入文本模式,关闭x server。
sudo service lightdm stop
sudo init 3
切换NVIDIA安装包指定目录,赋予权限并进行安装
chmod +x NVIDIA-Linux-x86_64-396.18.run
sudo sh NVID