Ubuntu 18.04 + Python3.6 + CUDA-10.0 + CUDNN-7.5.0 + tensorflow-gpu-1.13.1

Install nvidia driver

首先卸载原有驱动

sudo apt-get purge nvidia

添加图形驱动程序到源列表:

    sudo add-apt-repository ppa:graphics-drivers/ppa
    sudo apt update
    sudo apt upgrade

检查将安装什么驱动程序:

ubuntu-drivers devices

root@boss-To-be-filled-by-O-E-M:/home/boss# ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001E04sv00007377sd00001730bc03sc00i00
vendor   : NVIDIA Corporation
driver   : nvidia-driver-430 - third-party free recommended
driver   : nvidia-driver-410 - third-party free
driver   : nvidia-driver-415 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

注意！！！虚拟机无法安装NVIDIA显卡驱动程序，虚拟机中只有虚拟显卡没有NVIDIA显卡，不能安装NVIDIA驱动，无论是在NVIDIA官网下载NVIDIA-Linux-x86_64-430.40.run还是ppa下载都是没用的，也无法安装TensorFlow-gpu，虚拟机只能用cpu版本的TensorFlow。如果是考虑“虚拟一个系统”来使用显卡的话，考虑容器化技术——Docker大家可以参考这位大佬的文章虚拟机上无法安装tensorflow-GPU版本

虚拟机驱动程序是VM，，，

== /sys/devices/pci0000:00/0000:00:0f.0 ==
modalias : pci:v000015ADd00000405sv000015ADsd00000405bc03sc00i00
vendor   : VMware
model    : SVGA II Adapter
driver   : open-vm-tools-desktop - distro free

自动安装最新的驱动程序

sudo ubuntu-drivers autoinstall

显示

This system doesn't support Secure Boot
Secure Boot not enabled on this system.
Done.

nvidia:
Running module version sanity check.

- Original module
  - No original module exists within this kernel
- Installation
  - Installing to /lib/modules/5.0.0-25-generic/updates/dkms/

nvidia-modeset.ko:
Running module version sanity check.

- Original module
  - No original module exists within this kernel
- Installation
  - Installing to /lib/modules/5.0.0-25-generic/updates/dkms/

nvidia-drm.ko:
Running module version sanity check.

- Original module
  - No original module exists within this kernel
- Installation
  - Installing to /lib/modules/5.0.0-25-generic/updates/dkms/

nvidia-uvm.ko:
Running module version sanity check.

- Original module
  - No original module exists within this kernel
- Installation
  - Installing to /lib/modules/5.0.0-25-generic/updates/dkms/

depmod...

DKMS: install completed.

表示安装成功

然后重启机器:

sudo reboot

检查正确的安装驱动程序:

nvidia-smi

显示 说明NVIDIA驱动安装成功

root@boss-To-be-filled-by-O-E-M:/usr/local/cuda-10.0/samples/1_Utilities/bandwidthTest# nvidia-smi
Tue Aug 27 20:03:00 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.40       Driver Version: 430.40       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+=================