Advertisement

排查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)

还没有任何评论哟~