ubuntu18.04安装tensorflow-gpu2.1

1. 安装NIVIDA显卡驱动。

首先去官网下载NVIDIA显卡驱动，官网地址：https://www.nvidia.cn/Download/index.aspx?lang=cn

我下载的是目前最新的驱动：

我的显卡信息是：lspci | grep -i nvidia

使用wget下载.run文件：NVIDIA-Linux-x86_64-450.57.run

sudo ./NVIDIA-Linux-x86_64-450.57.run运行，选项全部点yes。

安装完后输入nvcc -V来检验是否安装成功：

2、利用conda安装tensorflow-gpu

命令：conda create -n tfgpu tensorflow-gpu=2.1

tfgpu为环境的名字，这个命令conda会自动安装对应的python, cudatoolkit, cudnn，不需要自己配置。

# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                        main    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
_tflow_select             2.1.0                       gpu    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
absl-py                   0.9.0                    py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
astor                     0.8.1                    py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
blas                      1.0                         mkl    https://mirrors.ustc.edu.cn/anaconda/pkgs/free
blinker                   1.4                      py36_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
c-ares                    1.15.0            h7b6447c_1001    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ca-certificates           2020.7.22                     0    https://mirrors.tu