【问题标题】:Blas GEMM launch failed in tensor flowBlas GEMM 在张量流中启动失败
【发布时间】:2021-12-04 22:32:57
【问题描述】:

您好,我正在尝试运行机器学习算法的代码

但是,当我运行代码时,我总是收到以下错误 Blas GEMM 启动失败

我使用 Geforce RTX 3070、CUDA 11.4 cuDNN 8.2。而Tensorflow 1.15.0和python 3.7版本的IDE是VSC

我以为这是 GPU 的内存问题,我写了一个代码来解决它,但它不起作用。此外,我删除了 Nvidia 中的缓存文件以节省更多空间,但它也没有工作

有没有解决这个问题的方法?还是这个问题与兼容性有关????

请帮助我天才堆栈溢出用户!!!

**Exception has occurred: InternalError
Blas GEMM launch failed : a.shape=(5000, 3), b.shape=(3, 20), m=5000, n=20, k=3
     [[node MatMul (defined at C:\Users\user\anaconda3\envs\tf\lib\site-packages\tensorflow_core\python\framework\ops.py:1748) ]]
Original stack trace for 'MatMul':
  File "C:\Users\user\anaconda3\envs\tf\lib\runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "C:\Users\user\anaconda3\envs\tf\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "c:\Users\user\.vscode\extensions\ms-python.python-2021.11.1422169775\pythonFiles\lib\python\debugpy\__main__.py", line 45, in <module>
    cli.main()
  File "c:\Users\user\.vscode\extensions\ms-python.python-2021.11.1422169775\pythonFiles\lib\python\debugpy/..\debugpy\server\cli.py", line 444, in main
    run()
  File "c:\Users\user\.vscode\extensions\ms-python.python-2021.11.1422169775\pythonFiles\lib\python\debugpy/..\debugpy\server\cli.py", line 285, in run_file
    runpy.run_path(target_as_str, run_name=compat.force_str("__main__"))
  File "C:\Users\user\anaconda3\envs\tf\lib\runpy.py", line 263, in run_path
    pkg_name=pkg_name, script_name=fname)**

【问题讨论】:

    标签: python tensorflow visual-studio-code


    【解决方案1】:

    是的,您的系统中TensorFlow-GPU 版本build configurations 的兼容性存在问题。

    您需要专门安装CUDA 10cudnn 7.4 以兼容在tensorflow_gpu-1.15.0Python 3.7 环境下运行您的代码。

    您可以在此链接中找到build configurations 的详细信息,并可以根据您的system OS 安装GPU support 要求。

    【讨论】:

      猜你喜欢
      • 2021-07-28
      • 2018-01-12
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-02-12
      • 2017-10-14
      • 1970-01-01
      • 2020-03-18
      相关资源
      最近更新 更多