docker ubuntu1804 opengl 支持 pytorch

最新推荐文章于 2023-04-06 02:13:12 发布

zdegrh_2

最新推荐文章于 2023-04-06 02:13:12 发布

阅读量831

点赞数

文章标签： docker

本文链接：https://blog.csdn.net/zdegrh_2/article/details/107005697

版权

本文提供了一个Dockerfile示例，用于在Ubuntu 18.04环境中搭建带有OpenGL支持的PyTorch开发环境。

摘要由CSDN通过智能技术生成

ARG from
 
FROM nvidia/opengl:1.0-glvnd-runtime-ubuntu18.04  as glvnd
FROM pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime

ARG DEBIAN_FRONTEND=noninteractive

RUN apt update && apt install -y \
		python3-tk \
		mlocate \
		curl \
		ca-certificates \
		pkg-config \
		p7zip-full

COPY --from=glvnd /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu
COPY --from=glvnd /usr/lib/i386-linux-gnu /usr/lib/i386-linux-gnu
 
COPY 10_nvidia.json /usr/share/glvnd/egl_vendor.d/10_nvidia.json
 
RUN echo '/usr/lib/x86_64-linux-gnu' >> /etc/ld.so.conf.d/glvnd.conf && \
    echo '/usr/lib/i386-linux-gnu' >> /etc/ld.so.conf.d/glvnd.conf && \
    ldconfig
 
ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:/usr/lib/i386-linux-gnu${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

ARG DEBIAN_FRONTEND=noninteractive
ENV NVIDIA_VISIBLE_DEVICES \
    ${NVIDIA_VISIBLE_DEVICES:-all}
ENV NVIDIA_DRIVER_CAPABILITIES \
    ${NVIDI