Skip to content

Commit 6c91e36

Browse files
committed
2 parents d473aa2 + d91eeb7 commit 6c91e36

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

commons/Dockerfile.base.py36-cuda10.1

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -260,9 +260,9 @@ RUN python3 -m pip install pip --no-cache-dir \
260260
notebook==6.0.3
261261

262262
WORKDIR /tmp
263-
COPY ./requirements.py38.txt /tmp
264-
RUN python3 -m pip install --no-cache-dir -r requirements.py38.txt && \
265-
rm -f /tmp/*.whl /tmp/requirements.py38.txt
263+
COPY ./requirements.txt /tmp
264+
RUN python3 -m pip install --no-cache-dir -r requirements.txt && \
265+
rm -f /tmp/*.whl /tmp/requirements.txt
266266

267267
# OpenCV
268268
RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
@@ -434,7 +434,6 @@ RUN jupyter nbextensions_configurator enable && \
434434
jupyter lab build
435435

436436
RUN apt autoclean && \
437-
ln -s /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcusolver.so.10 /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcusolver.so.10 && \
438437
sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
439438
python3 -m pip uninstall -y tensorboard-plugin-wit && \
440439
rm -rf /var/lib/apt/lists/* && \

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)