gpt4 book ai didi

docker - 在docker上服务的TensorFlow对cuInit的调用失败:CUresult(-1)

转载 作者:行者123 更新时间:2023-12-02 21:15:03 26 4
gpt4 key购买 nike

在Kuberenetes上使用图像tensorflow/serving:latest-devel-gpu时,未使用GPU。

我对此没有任何幻想,只需传递server.conf和模型文件即可。

默认运行时为nvidia-docker,我的其他GPU pods 能够使用GPU。

日志中唯一的错误:

E external/org_tensorflow/tensorflow/stream_executor/cuda/cuda_driver.cc:397 ] failed call to cuInit: CUresult(-1)



有趣的是:

I external/org_tensorflow/tensorflow/stream_executor/cuda/cuda_diagnostics.cc:189] libcuda reported version is: Not found: was unable to find libcuda.so DSO loaded into this program

最佳答案

用更新你的dockerfile

RUN rm /usr/local/cuda/lib64/stubs/libcuda.so.1

或从docker hub扩展提供服务的devel gpu镜像,额外增加一行
FROM tensorflow/serving:1.9.0-devel-gpu
RUN rm /usr/local/cuda/lib64/stubs/libcuda.so.1

关于docker - 在docker上服务的TensorFlow对cuInit的调用失败:CUresult(-1),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51522247/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com