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降级