Ubuntu 18.04 安装tensorflow gpu版本

背景

系统：Ubuntu 18.04
显卡：rtx 2060
cuda版本：cuda10.2

Ubuntu 18.04安装cuda10.2和CUDNN8.2.2

见我另外一篇博客：Ubuntu 18.04 安装NVIDIA显卡驱动+cuda 10.2+cudnn

重新安装tensorflow-GPU步骤

创建环境

conda create --name tensorflow_gpu python=3.6

激活环境

activate tensorflow_gpu

查看环境

conda info --envs

安装tensorflow

conda install tensorflow-gpu==1.14.0

安装keras

pip install keras==2.2.5

安装其他库

pip install Pillow
pip install matplotlib
pip install opencv-python
pip install PyQt5

Pycharm增加一个环境

目前有了两个环境，增加一个刚才新建那个，如下图

测试成功