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"]
|
