test-nvidia/assets/Dockerfile

22 lines
561 B
Text
Raw Normal View History

2020-09-05 12:59:40 +02:00
ARG BASE_IMAGE
FROM $BASE_IMAGE
2021-10-20 11:14:34 +02:00
ARG tensorflow_package
ARG tensorflow_version
ARG test_nvidia_options
2020-09-05 12:59:40 +02:00
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
2021-10-20 11:14:34 +02:00
RUN pip install --no-cache-dir "$tensorflow_package == $tensorflow_version"
2020-09-05 12:59:40 +02:00
COPY test-nvidia /usr/bin
2021-10-20 11:14:34 +02:00
ENV test_nvidia_options $test_nvidia_options
CMD /usr/bin/test-nvidia $test_nvidia_options