✨ Test TensorFlow 2
This commit is contained in:
parent
cfeafeb57f
commit
5fa9406fbd
8 changed files with 68 additions and 29 deletions
19
assets/Dockerfile
Normal file
19
assets/Dockerfile
Normal 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"]
|
1
assets/requirements-tf1.txt
Normal file
1
assets/requirements-tf1.txt
Normal file
|
@ -0,0 +1 @@
|
|||
tensorflow-gpu == 1.15.*
|
1
assets/requirements-tf2.txt
Normal file
1
assets/requirements-tf2.txt
Normal file
|
@ -0,0 +1 @@
|
|||
tensorflow == 2.*
|
Loading…
Add table
Add a link
Reference in a new issue