zoukankan      html  css  js  c++  java
  • 运行TensorFlow代码时报错

    运行TensorFlow代码时报错

    错误信息ImportError: libcublas.so.10.0: cannot open shared object file

    原因:TensorFlow版本与CUDA版本不匹配

    可使用pip3 install tensorflow-gpu==[version]将TensorFlow版本切换到制定版本中去,其对应的version可在TensorFlow官网查看

    另:

    1.查看cuda版本(在ubuntu下):

    cat /usr/local/cuda/version.txt
    或者

    nvcc --version

    2.查看cudnn版本(在ubuntu下):

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    3.查看tensorflow-gpu的版本:

    pip list

  • 相关阅读:
    正则表达式
    特殊符号作用
    sed
    scp
    EOF
    env
    JAVA进阶5
    JAVA进阶4
    JAVA进阶3
    JAVA进阶2
  • 原文地址:https://www.cnblogs.com/youpeng/p/10887389.html
Copyright © 2011-2022 走看看