Ubuntu18.04+Nvidia+Cuda10+TensorFlow-GPU2.0

Ubuntu18.04+Nvidia+Cuda10+TensorFlow2.0

---

一、Nvidia驱动安装

# 列出设备，显示的model是显卡型号，driver是可选的驱动
ubuntu-drivers devices
# 安装其中一个驱动
sudo apt install nvidia-340
sudo reboot

二、安装Cuda

1. 下载Cuda

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu
我这里下载的是Cuda10.0

需要注意的是，之前已经安装过显卡驱动程序，在提问是否安装显卡驱动时选择no
其他 选择默认路径或者yes即可。

2. 如果之前已经安装了Cuda，则卸载掉。

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0

3. 安装

chomd +x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run

三、安装cuDNN

https://developer.nvidia.com/rdp/cudnn-download
这里下载 deb 格式，安装：

 sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb

四、安装pip tensorflow

sudo apt install python3-pip -y
pip3 install tensorflow-gpu -i https://pypi.douban.com/simple/