While programming with python3.6 on DGX Station (NVIDIA) based on ONNX runtime environment;
Using following libraries;
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
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
runtime: nvidia
and all started working
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.