Skip to content

Commit 87ea992

Browse files
committed
add service-defs
1 parent f30abfa commit 87ea992

File tree

1 file changed

+27
-22
lines changed

1 file changed

+27
-22
lines changed

python-tensorflow/Dockerfile.2.4-py38-cuda11.1

Lines changed: 27 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -24,40 +24,44 @@ RUN python3 -m pip install --no-cache-dir \
2424

2525
RUN python3 -m pip install --no-cache-dir \
2626
--use-feature=2020-resolver \
27-
mesh-tensorflow==0.1.13 \
28-
google_cloud_spanner==1.19.0 \
29-
pymongo==3.11.2 \
30-
googleapis-common-protos==1.51.0 \
31-
google_cloud_videointelligence==1.15.0 \
32-
google_cloud_vision==0.42.0 \
33-
google_cloud_pubsub==1.6.0 \
34-
google_cloud_dlp==0.15.0 \
35-
google_cloud_datastore==1.15.2 \
3627
fastavro==1.2.0 \
37-
google_cloud_bigtable==1.6.0 \
38-
google_api_core==1.24.0 \
28+
pymongo==3.11.2 \
29+
docker==4.4.1 \
3930
pbr==5.5.1 \
4031
mock==1.3.0 \
4132
proto-plus==1.13.0 \
4233
oauth2client==4.1.3 \
34+
future==0.18.1 \
35+
gin_config==0.4.0 \
4336
joblib==0.14.0 \
4437
libcst==0.3.14
4538

4639
RUN python3 -m pip install --no-cache-dir \
4740
--use-feature=2020-resolver \
41+
google_cloud_spanner==1.19.0 \
42+
googleapis-common-protos==1.51.0 \
43+
google_cloud_videointelligence==1.15.0 \
44+
google_cloud_vision==0.42.0 \
45+
google_cloud_pubsub==1.6.0 \
46+
google_cloud_dlp==0.15.0 \
47+
google_cloud_datastore==1.15.2 \
48+
google_cloud_bigtable==1.6.0 \
49+
google_api_core==1.24.0
50+
51+
RUN python3 -m pip install --no-cache-dir \
52+
mesh-tensorflow==0.1.18 \
4853
tensorflow-cloud==0.1.10 \
49-
tensorflow-hub==0.9.0 \
50-
tensorflow_text==2.4.1 \
51-
tensorflow-graphics-gpu==1.0.0 \
52-
tensorflow-gan==2.0.0
54+
tensorflow-hub==0.9.0
55+
5356
RUN python3 -m pip install --no-cache-dir \
54-
--use-feature=2020-resolver \
55-
tensorflow-data-validation \
56-
tensorflow-plot==0.3.2 \
57-
# tensorflow-lattice==2.0.7 \
57+
-i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation
58+
RUN python3 -m pip install --no-cache-dir \
59+
tensorflow-graphics-gpu==1.0.0 \
60+
tensorflow-io==0.17.0 \
5861
tensorflow_ranking==0.3.2 \
59-
# tensorflow-compression==1.3 \
60-
tensorflow-io==0.17.0
62+
tensorflow-plot==0.3.2 \
63+
tensorflow_text==2.4.1 \
64+
tensorflow-gan==2.0.0
6165

6266
RUN python3 -m pip install --extra-index-url \
6367
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100 \
@@ -78,7 +82,8 @@ RUN python3 -m pip install --no-cache-dir \
7882
nni==1.9 \
7983
mlflow==1.12.1 \
8084
scikit-nni==0.2.1
81-
85+
86+
COPY ./service-defs /etc/backend.ai/service-defs
8287
# Install ipython kernelspec
8388
Run python3 -m ipykernel install --display-name "TensorFlow 2.4 on Python 3.8 & CUDA 11ᇂᇂᇂᇂ.1" && \
8489
cat /usr/local/share/jupyter/kernels/python3/kernel.json

0 commit comments

Comments
 (0)