Ubuntu18.04 TensorFlow2.0安装及配置

环境配置要求

• python 3.6
• NVIDIA® GPU drivers —CUDA 10.0 requires 410.x or higher.
• CUDA® Toolkit —TensorFlow supports CUDA 10.0 (TensorFlow >= 1.13.0)
• CUPTI ships with the CUDA Toolkit.
• cuDNN SDK (>= 7.4.1)
• (Optional) TensorRT 5.0 to improve latency and throughput for inference on some models.

NVIDIA驱动安装

1. 准备工作
   - 禁用BIOS中的secure boot，因为此方法使用第三方源安装显卡驱动，不禁止secure boot会导致安装的驱动不能使用，禁用也不会有多大安全隐患。
   - 禁用nouveau，这是ubuntu默认使用的开源显卡驱动，和nvidia驱动一起使用可能导致黑屏，所以禁掉。
2. 禁用nouveau
   创建blacklist-nouveau.conf

   $ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
   

   将下述内容插入该文件

   blacklist nouveau
   options nouveau modeset=0
   

   执行以下命令使禁用生效并且重启：

   $ sudo update-initramfs -u
   $ sudo reboot