解决Docker镜像报错的问题:无法打开共享对象文件libcuda.so.1不存在
发布时间
阅读量:
阅读量
问题描述
获取包含Python3环境且安装有tensorflow-gpu==1.13.1版本的镜像
docker pull tensorflow/tensorflow:1.13.1-gpu-py3
确定特定的容器
docker run -it -v /xxx/xxx/xxx:/xxx/xxx --name tf-113 xxxxxxxxxxxx /bin/bash
在进入镜像环境后,通过执行pip命令进行查询,确认tensorflow-gpu=1.13.1版本已成功完成安装。
pip list
然而在调用tensorflow过程中出现了错误提示
ImportError: libcuda.so.1: cannot open shared object file: No such file or directory
问题分析
libcuda报错提示表明当前使用的镜像缺少对应的显卡驱动支持,最初误以为需要重新安装适配版本的cudatoolkit与cudnn库,随后才认识到问题根源在于运行镜像时未正确配置GPU相关参数。
解决方法
对执行命令进行调整,增加“--gpus all”参数后即可解决问题。
全部评论 (0)
还没有任何评论哟~
