Skip to content

Instantly share code, notes, and snippets.

@flx42
Created August 23, 2017 13:35
Show Gist options
  • Save flx42/91a501dcd7f64a0845808b99f2a00d06 to your computer and use it in GitHub Desktop.
Save flx42/91a501dcd7f64a0845808b99f2a00d06 to your computer and use it in GitHub Desktop.
docker multi-stage deviceQuery
FROM nvidia/cuda:8.0-devel-ubuntu16.04 as build
RUN apt-get update && apt-get install -y cuda-samples-8-0
RUN make -C /usr/local/cuda/samples/1_Utilities/deviceQuery
FROM ubuntu:16.04
COPY --from=build /usr/local/cuda/samples/1_Utilities/deviceQuery/deviceQuery /usr/local/bin/deviceQuery
LABEL com.nvidia.volumes.needed="nvidia_driver"
ENV PATH /usr/local/nvidia/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
CMD deviceQuery
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment