Blas GEMM 在张量流中发射失败

Question

嗨，我正在尝试运行机器学习算法的代码

但是，当我运行代码时，我总是收到以下错误 Blas GEMM launch failed

我使用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)**

Answer 1

是的，您的系统

TensorFlow-GPU

版本

build configurations

兼容性有问题。

您需要专门安装

CUDA 10

和

cudnn 7.4

才能兼容在

tensorflow_gpu-1.15.0

和

Python 3.7

环境中运行您的代码。

您可以在此链接中找到构建配置详细信息，并可以根据您的system OS安装

GPU支持

要求。

Answer 2

旧线程，但看起来几乎不可能将 pre 2.x 张量流（特别是 tf 1.15）与 Amper 架构 Nvidia GPU 配对，希望我错了。