解决Tensorflow error(CUBLAS_STATUS_EXECUTION_FAILED)报错问题

最新推荐文章于 2024-05-13 23:55:20 发布

Zpadger

最新推荐文章于 2024-05-13 23:55:20 发布

阅读量2.6k

点赞数 4

分类专栏： Python debug 踩坑记录文章标签： tensorflow python CUDA RTX显卡

本文链接：https://blog.csdn.net/spadgerz/article/details/127547662

版权

在Ubuntu20.04系统和RTX A5000显卡环境中，使用Tensorflow-gpu==1.13.1时遇到CUBLAS_STATUS_EXECUTION_FAILED错误。尝试升级到1.14.0版本无效，因为RTX30系列显卡需要CUDA 11.1及以上版本。解决方案是重新编译CUDA 11版本的tensorflow-gpu==1.15.0。

摘要由CSDN通过智能技术生成

问题描述

系统：Ubuntu20.04

显卡：RTX A5000

物理机CUDA Verion：11.4

所安装的Tensorflow版本：tensorflow-gpu==1.13.1

采用Conda创建了虚拟环境以及拉取Tensorflow官方docker镜像两种方式，

其中Conda虚拟环境配置（Python3.7.2），使用conda命令安装，如下所示：

cudatoolkit               10.0.130            h8c5a6a4_10    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
cudnn                     7.6.5.32             ha8d7eb6_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

事实证明tensorflow-gpu==1.13.1是可以在该虚拟环境下运行成功：

>>> tf.test.is_gpu_available()
2022-10-27 10:52:22.010425: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 AVX512F FMA
2022-10-27 10:52:22.059681: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 3400000000 Hz
2022-10-27 10:52:22.063477: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x55c559aae5b0 executing computations on platform Host. Devices:
2022-10-27 10:52:22.063507: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (0): <undefined>, <undefined>
2022-10-27 10:52:22.066212: I tensorflow/stream_executor/platform/default/dso_loader.cc:42] Successfully opened dynamic library libcuda.so.1
2022-10-27 10:52:22.399495: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x55c55ba1eed0 executing computations on platform CUDA. Device