在自定义docker映像中运行tensorflow时发生错误

时间:2020-10-22 08:34:32

标签: docker tensorflow

我正在尝试在Codalab上提交论文。

当我使用给定here的tesnorflow docker映像时,看到没有错误。但是,Docker映像中缺少许多库,因此我制作了一个自定义Docker映像here

两个文件几乎相同。 但是,当我运行自定义构建的图像时,仍然出现这样的错误

2020-10-22 08:21:46.257137: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] 
Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open 
shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:
/usr/local/nvidia/lib:/usr/local/nvidia/lib64

引起问题的代码行是

ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH

这行如何导致我的映像而不是tensorflow docker映像出现问题?

这是成功构建dockerhub上仓库的link

0 个答案:

没有答案