cuda相关的dockerfile build时异常处理

最新推荐文章于 2024-07-29 14:55:16 发布

日后定负责

最新推荐文章于 2024-07-29 14:55:16 发布

阅读量976

点赞数

本文链接：https://blog.csdn.net/matrix273/article/details/107880129

版权

ubuntu 同时被 3 个专栏收录

14 篇文章 1 订阅

订阅专栏

docker

5 篇文章 0 订阅

订阅专栏

cuda

2 篇文章 0 订阅

订阅专栏

在测试gpuburn的docker镜像时，使用以下dockerfile，

FROM nvidia/cuda:9.0-devel

RUN apt-get update && \
apt-get install -y --no-install-recommends \
    wget \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /root
RUN wget http://wili.cc/blog/entries/gpu-burn/gpu_burn-1.1.tar.gz \
    && tar xzf gpu_burn-1.1.tar.gz \
    && make

ENTRYPOINT [ "/root/gpu_burn" ]
CMD [ "10" ]

在build时提示报错

E: Failed to fetch https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/Packages  Writing more data than expected (1309930 > 1308522)
E: Some index files failed to download. They have been ignored, or old ones used instead.
The command '/bin/sh -c apt-get update &&     apt-get install -y --no-install-recommends     wget     && rm -rf /var/lib/apt/lists/*' returned a non-zero code: 100

经搜索得知需要删除以前的 cuda相关的源sourlist文件,即在 RUN apt-get update 前执行以下操作

RUN rm /etc/apt/sources.list.d/cuda.list && rm /etc/apt/sources.list.d/nvidia-ml.list

最终的dockerfile如下

FROM nvidia/cuda:9.0-devel

RUN rm /etc/apt/sources.list.d/cuda.list && rm /etc/apt/sources.list.d/nvidia-ml.list
RUN apt-get update && \
apt-get install -y --no-install-recommends \
    wget \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /root
RUN wget http://wili.cc/blog/entries/gpu-burn/gpu_burn-1.1.tar.gz \
    && tar xzf gpu_burn-1.1.tar.gz \
    && make

ENTRYPOINT [ "/root/gpu_burn" ]
CMD [ "10" ]

然后就可以使用docker build 构建镜像了