Skip to content

Commit d91eeb7

Browse files
committed
deduplicate tensorflow
1 parent 324af0f commit d91eeb7

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

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

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,15 @@ RUN python3 -m pip install --no-cache-dir \
6262
tensorflow-plot==0.3.2 \
6363
tensorflow_text==2.4.1 \
6464
tensorflow-gan==2.0.0
65-
65+
RUN python3 -m pip uninstall --no-cache-dir -y \
66+
tensorflow==2.3.0 \
67+
tensorflow-gpu==2.4.1 \
68+
tensorboard==2.3.0 \
69+
tensorboard-plugin-wit==1.7.0 && \
70+
python3 -m pip install --no-cache-dir \
71+
tensorflow-gpu==2.4.1 \
72+
tensorboard-plugin-wit==1.8.0 \
73+
tensorboard==2.4.0
6674
RUN python3 -m pip install --extra-index-url \
6775
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100 \
6876
&& \

0 commit comments

Comments
 (0)