Blas GEMM发射在张量流中失败

Question

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

但是，当我运行代码时，总是会出现以下错误: Blas启动失败

我使用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支持需求。