源码编译安装tensorflow GPU版本过程

tensorflow-build

源码编译安装tensorflow GPU版本过程

硬件信息及版本

• hw ：
  • Intel® Core™ i7-6700HQ CPU @ 2.60GHz
  • NVIDIA GTX1060 6G GDDR5
  • 8GB DDR4内存
• OS ： Ubuntu 18.04.2 LTS
• tensorflow branch : origin/r1.13
• cuda : 10.0
• cudnn: 7.4.2
• nccl: 2.4.2
• python ：2.7
• bazel: 0.19.2

+ 源码编译安装TensorFlow版本需要对应，不然可能会出现不可预期的错误！

查看GPU信息

命令

• lspci | grep -i nvidia
• sudo lshw -numeric -C display

输出

• 06:00.0 VGA compatible controller: NVIDIA Corporation GK110B [GeForce GTX TITAN Z] (rev a1)
• 06:00.1 Audio device: NVIDIA Corporation GK110 HDMI Audio (rev a1)
• 07:00.0 3D controller: NVIDIA Corporation GK110B [GeForce GTX TITAN Z] (rev a1)

禁用nouveau驱动

  + 使用核显和独显，因为对NVIDIA显卡驱动支持不太好，很多在安装TensorFlow GPU版本时经常会遇到黑屏，进不到Linux系统中等问题； 因此在安装显卡驱动和CUDA前先禁用nouveau驱动

操作

• sudo vim /etc/modprobe.d/blacklist-nouveau.conf
• # 输入以下内容
• blacklist nouveau
• options nouveau modset=0

执行

• sudo update-initramfs -u

重启电脑查看是否成功

• sudo lspci | grep nouveau
  如果没有内容，则禁用成功。

安装cuda