排查Tensorflow错误(CUBLAS_STATUS_EXECUTION_FAILED)导致程序崩溃
发布时间
阅读量:
阅读量
问题描述
系统环境:Ubuntu20.04
显卡型号:RTX A5000
物理机所使用的CUDA版本: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.
全部评论 (0)
还没有任何评论哟~
