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