[英]CUDA libcublas.so.11 Error when using GPUs inside ONNX Docker Container
While programming with python3.6 on DGX Station (NVIDIA) based on ONNX runtime environment;在基于ONNX运行环境的DGX Station(NVIDIA)上使用python3.6编程时;
Using following libraries;使用以下库;
mxnet==1.5.x onnxruntime-gpu==1.7.x mxnet==1.5.x onnxruntime-gpu==1.7.x
I see following error我看到以下错误
OSError: libcublas.so.11: cannot open shared object file: No such file or directory
Fixed it by updating CUDA to 11.0通过将 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
then a new error started appearing然后一个新的错误开始出现
onnxruntime cuda failure 100 no cuda-capable device is detected
then it was noted that docker was not started with nvidia runtime so following was added to the docker-compose file然后注意到 docker 未使用 nvidia 运行时启动,因此将以下内容添加到 docker-compose 文件中
runtime: nvidia
and all started working一切都开始工作了
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.