mikegerber/test-nvidia
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

⬆️ Update for TensorFlow 2.5+2.6

Browse source
This commit is contained in:
Gerber, Mike 2021-10-20 11:14:34 +02:00
parent 5e20d251d6
commit 5d05953b5f
6 changed files with 77 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

10
assets/Dockerfile
View file

@ -2,7 +2,9 @@ ARG BASE_IMAGE
FROM $BASE_IMAGE
ARG tf
ARG tensorflow_package
ARG tensorflow_version
ARG test_nvidia_options
RUN apt-get update && \
apt-get install -y python3 python3-distutils curl &&\
@ -11,9 +13,9 @@ RUN apt-get update && \
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
RUN pip install --no-cache-dir "$tensorflow_package == $tensorflow_version"
COPY test-nvidia /usr/bin
CMD ["/usr/bin/test-nvidia"]
ENV test_nvidia_options $test_nvidia_options
CMD /usr/bin/test-nvidia $test_nvidia_options