[英]“ImportError: libcublas.so.10.0” for Tensorflow ONLY when using GUI
[英]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.