tensorflow cuda cudnn

绝地灬酷狼 2021-11-27 05:06 526阅读 0赞

  • tf版本问题

    问题： tensorflow==1.12.0 提供非阻塞服务时，报错tensorflow/core/common_runtime/bfc_allocator.cc:380] Check failed: h != kInvalidChunkHandle；另一方面，阻塞服务时运行正常
    解决方法： 替换为1.13rc0后运行正常

  • gpu model及其编译结构



















gpu model architecture cuda
Tesla K80 SM37 or SM_37 CUDA 5 and later
Tesla V100 SM70 or SM_70 CUDA 9 and later

  • cuda 驱动版本查看

    cat /proc/driver/nvidia/version

  • cuda编译器版本查看

    nvcc —version

    cat /usr/local/cuda/version.txt

  • cuDNN与cuda版本对应关系
















cuDNN cuda
7.6.1 9.0，9.2，10.0，10.1
5.1 8.0，7.5

  • cuDNN版本查看

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2 //或者试试这个

  • tensorflow测试的构建配置

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2h6aGoyMDA3_size_16_color_FFFFFF_t_70

  • pytorch与cuda版本的对应

参考文献：

  1. check-cuda-installed
  2. Supported SM and Gencode variations
  3. cuDNN Archive
  4. tf源码编译
  5. Which TensorFlow and CUDA version combinations are compatible?
  6. https://pytorch.org/get-started/previous-versions/

发表评论

表情：
评论列表 （有 0 条评论，526人围观）

还没有评论，来说两句吧...

相关阅读