1.驱动安装(nvidia-smi)

2.cuda安装(nvcc -V)

3.cudnn安装(cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2)

4.gcc/g++版本(gcc -v)

gcc升降级(可选)

sudo apt-get install -y gcc-4.8
sudo apt-get install -y g++-4.8
cd /usr/bin
sudo rm gcc
sudo ln -s gcc-4.8 gcc
sudo rm g++
sudo ln -s g++-4.8 g++

5.anaconda(conda -V)

创建虚拟环境(pyhton版本为3.6),安装tensorflow。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.9.0

测试tensorflow

6.安装运行需要python库

cython,opencv-python,easydict,Pillow,matplotlib,scipy,pyyaml

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cython

7.pycharm(IDE)

设置编译器为conda环境

8.补充

 pip install numpy==1.16#numpy降级