You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
327 B
Docker
11 lines
327 B
Docker
FROM nvidia/cuda:latest
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,video,utility
|
|
|
|
RUN mkdir -p /usr/local/bin /patched-lib
|
|
COPY patch.sh docker-entrypoint.sh /usr/local/bin/
|
|
RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh
|
|
|
|
ENTRYPOINT ["/usr/local/bin/docker-entrypoint.sh"]
|