本文主要讲述基于Ubuntu18.04LTS 版本的tenserflow1.8安装
配置如下:
Ubuntu 18.04 LTS
python 2.7
nvidia driver 390
cuda 9.0
cudnn 7.1
tensorflow 1.8
- 安装NVIDIA driver
查看NVIDIA driver 推荐。
$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C81sv00001028sd000011C0bc03sc00i00
vendor : NVIDIA Corporation
model : GP107 [GeForce GTX 1050]
driver : nvidia-driver-396 - third-party free recommended
driver : nvidia-driver-390 - third-party free
driver : xserver-xorg-video-nouveau - distro free builtin
可以看到当前推荐的是nvidia-driver-396- third-party free。
安装:
$ sudo ubuntu-drivers autoinstall按照推荐的驱动安装,如果失败,请添加NVIDIA driver源仓库然后在安装,安装命令如下:
$ sudo add-apt-repository ppa:graphics-drivers/ppa
测试安装是否成功
$ nvidia-smi