006 CUDA/CUDNN与tensorflow

2019-04-14  本文已影响0人  Oscar_hailiang
1. 查看cuda版本
nvcc -V
或:cat /usr/local/cuda/version.txt
2. 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
或:cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
  1. Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
碰到的几种坑:
  1. 问题: tensorflow的CUDA driver version is insufficient for CUDA runtime
    https://blog.csdn.net/li57681522/article/details/82491617
  2. 问题:could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
    https://blog.csdn.net/xianqin_ma/article/details/79525519
    cuda版本升级:
    https://blog.csdn.net/wanzhen4330/article/details/81704474
上一篇 下一篇

猜你喜欢

热点阅读