Ubuntu下配置TensorFlow-Gpu版,其对应CUDA、cuDNN版本

TensorFlow官网安装指导界面提示“运行支持 GPU 的 TensorFlow 所需满足的 NVIDIA 要求” 查看机器CUDA、cuDNN版本命令分别是: cat /usr/local/cuda/version.txt cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2  例如得到如下信息: CUDA Version 8
相关文章
相关标签/搜索