代码之家  ›  专栏  ›  技术社区  ›  Yuwen Yan

使用旧版本的CUDA和CUDN安装GPU版本TensorFlow[重复]

  •  3
  • Yuwen Yan  · 技术社区  · 8 年前

    这个问题已经有了答案:

    目前,我需要在一台存在旧版本CUDA和CUDN的机器上安装GPU版本的TensorFlow,我的问题是如何在不重新安装CUDA和CUDN的情况下完成它,因为其他包(请参阅pytorch)需要它。

    1 回复  |  直到 8 年前
        1
  •  5
  •   Yuwen Yan    8 年前

    根据以下命令检查CUDA版本 this post .

    $ nvcc --version
    

    根据以下命令检查CUDNN版本 this post .

    $ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    

    根据选择TensorFlow GPU版本 this post

    enter image description here

    推荐文章