繁体   English   中英

CUDA libcublas.so.11 在 ONNX Docker 容器内使用 GPU 时出错

[英]CUDA libcublas.so.11 Error when using GPUs inside ONNX Docker Container

在基于ONNX运行环境的DGX Station(NVIDIA)上使用python3.6编程时;

使用以下库;

mxnet==1.5.x onnxruntime-gpu==1.7.x

我看到以下错误

OSError: libcublas.so.11: cannot open shared object file: No such file or directory

通过将 CUDA 更新为 11.0 来修复它

apt update
apt install cuda-11.0
pip3 uninstall mxnet //and reinstall latest version
pip3 uninstall onnxruntime-gpu //and resinstall latest version

然后一个新的错误开始出现

onnxruntime cuda failure 100 no cuda-capable device is detected

然后注意到 docker 未使用 nvidia 运行时启动,因此将以下内容添加到 docker-compose 文件中

runtime: nvidia

一切都开始工作了

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM