输入命令:
pip install tensorflow-gpu==1.4.1
ps:cuda:8.0
cudnn:6.0
cuda版本查看:
cat /usr/local/cuda/version.txt
cudnn 版本查看:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
输入命令:
pip install tensorflow-gpu==1.4.1
ps:cuda:8.0
cudnn:6.0
cuda版本查看:
cat /usr/local/cuda/version.txt
cudnn 版本查看:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2