Test TensorFlow 2

This commit is contained in:
Gerber, Mike 2020-09-05 12:59:40 +02:00
parent cfeafeb57f
commit 5fa9406fbd
8 changed files with 68 additions and 29 deletions

19
assets/Dockerfile Normal file
View file

@ -0,0 +1,19 @@
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"]

View file

@ -0,0 +1 @@
tensorflow-gpu == 1.15.*

View file

@ -0,0 +1 @@
tensorflow == 2.*