首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >CUDA libcublas.so.11在ONNX码头容器内使用GPU时出错

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

Stack Overflow用户
提问于 2022-02-02 17:16:44
回答 1查看 469关注 0票数 1

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

使用下列图书馆;

mxnet==1.5.x 1.7.x运行时-gpu=1.7.x

我看到下面的错误

代码语言:javascript
复制
OSError: libcublas.so.11: cannot open shared object file: No such file or directory
EN

回答 1

Stack Overflow用户

发布于 2022-02-02 17:16:44

修正了将数据自动化系统更新为11.0

代码语言:javascript
复制
apt update
apt install cuda-11.0
pip3 uninstall mxnet-cu92 //and reinstall mxnet-cu102
pip3 uninstall onnxruntime-gpu //and resinstall latest version

接着,一个新的错误出现了。

代码语言:javascript
复制
onnxruntime cuda failure 100 no cuda-capable device is detected

然后注意到,没有使用nvidia运行时启动docker,因此将下面的内容添加到docker-复合文件中。

代码语言:javascript
复制
runtime: nvidia

所有人都开始工作了

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70959943

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档