1、查看cuda版本

cat /usr/local/cuda/version.txt

2、查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

3.切换cudnn

1 sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
2 sudo cp cuda/include/cudnn.h /usr/local/cuda/include
3 sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
4 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

4.切换cuda

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda

增加源

sudo pip3 install tensorflow-gpu==1.4 -i https://pypi.tuna.tsinghua.edu.cn/simple