Ubuntu18.04 TensorFlow2.0-GPU安装及配置
环境配置要求
- 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驱动安装
- 准备工作
- 禁用BIOS中的secure boot,因为此方法使用第三方源安装显卡驱动,不禁止secure boot会导致安装的驱动不能使用,禁用也不会有多大安全隐患。
- 禁用nouveau,这是ubuntu默认使用的开源显卡驱动,和nvidia驱动一起使用可能导致黑屏,所以禁掉。 - 禁用nouveau
创建blacklist-nouveau.conf
将下述内容插入该文件$ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
执行以下命令使禁用生效并且重启:blacklist nouveau options nouveau modeset=0
$ sudo update-initramfs -u $ sudo reboot