cuda相关的dockerfile build时异常处理
在测试gpuburn的docker镜像时，使用以下dockerfile，FROM nvidia/cuda:9.0-develRUN apt-get update && \apt-get install -y --no-install-recommends \wget \&& rm -rf /var/lib/apt/lists/*WORKDIR /rootRUN wg
·
在测试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 构建镜像了
docker build -t gpuburn .
欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)