Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,

关于这个错误，这个帖子给出了完善的解释https://stackoverflow.com/questions/53698035/failed-to-get-convolution-algorithm-this-is-probably-because-cudnn-failed-to-in对于我个人情况，加入以下代码就行os.environ["CUDA_DEVICE_ORDER"] = "PCI_BU

ndwuhuangwei

877人浏览 · 2022-03-29 16:58:35

ndwuhuangwei · 2022-03-29 16:58:35 发布

关于这个错误，这个帖子给出了完善的解释
https://stackoverflow.com/questions/53698035/failed-to-get-convolution-algorithm-this-is-probably-because-cudnn-failed-to-in

对于我个人情况，加入以下代码就行

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"  

gpus = tf.config.experimental.list_physical_devices("GPU")
if gpus:
    try:
        for gpu in gpus:
            tf.config.experimental.set_memory_growth(gpu, True)
    except RuntimeError as e:
        print(e)
        exit(-1)

昇腾开源生态专区

昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐

Flutter 框架跨平台鸿蒙开发 - 开发二维码生成器与扫描器

size: 200,// 定位点样式（三个角的方块）eyeShape: QrEyeShape.square, // square 或 circle),// 数据点样式dataModuleShape: QrDataModuleShape.square, // square 或 circle),二维码生成：qr_flutter库的使用和样式自定义二维码扫描：mobile_scanner相机调用和结果处

昇腾开源生态专区

必学收藏！vLLM-Omni全模态模型高效推理引擎：从入门到实践

昇腾开源生态专区

PD分离部署：基于Atlas 800T A3服务器+SGLang 1P2D部署Qwen3-235B指南

Qwen3-235B模型用Atlas 800I A3或Atlas 800T A3均可部署，本文档以Atlas 800T A3为例，记录了使用SGLang框架部署Qwen3-235B模型的1P2D（一个Prefill节点和两个Decode节点）配置方案。该配置针对大规模语言模型推理场景进行了优化，特别适用于需要高吞吐量的生产环境。