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.
20 lines
458 B
Docker
20 lines
458 B
Docker
ARG BASE_IMAGE
|
|
|
|
FROM $BASE_IMAGE
|
|
|
|
ARG tf
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y python3 python3-distutils curl &&\
|
|
apt-get clean && rm -rf /var/lib/apt/lists/* && \
|
|
\
|
|
curl -sSL https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
|
|
python3 get-pip.py && \
|
|
rm -f get-pip.py
|
|
COPY assets/requirements-$tf.txt /tmp
|
|
RUN pip install --no-cache-dir -r /tmp/requirements-$tf.txt
|
|
|
|
COPY test-nvidia /usr/bin
|
|
|
|
CMD ["/usr/bin/test-nvidia"]
|