Advertisement

解决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)

还没有任何评论哟~