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
相關文章
相關標籤/搜索