test-nvidia/Dockerfile

12 lines
360 B
Text
Raw Normal View History

2019-10-15 12:46:51 +02:00
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
RUN apt-get update &&\
apt-get install -y python3 python3-pip &&\
apt-get clean && rm -rf /var/lib/apt/lists/*
COPY requirements.txt /tmp
2020-05-15 13:14:24 +02:00
RUN pip3 install --no-cache-dir --upgrade pip && \
pip3 install --no-cache-dir -r /tmp/requirements.txt
2019-10-15 12:46:51 +02:00
COPY test-nvidia /usr/bin
CMD ["/usr/bin/test-nvidia"]