簡體   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