解决Docker镜像报错“ImportError: libcuda.so.1: cannot open shared object file: No such file or directory”问题

问题描述

拉取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指令查看发现已经安装成功tensorlfow-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

解决方法

更换指令，添加“--gous all”，即可

docker run -it --gpus all -v /xxx/xxx/xxx:/xxx --name tf-113 xxxxxxxxxxxx /bin/bash

如图所示：

root@b8