File tree Expand file tree Collapse file tree 2 files changed +12
-5
lines changed Expand file tree Collapse file tree 2 files changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -260,9 +260,9 @@ RUN python3 -m pip install pip --no-cache-dir \
260
260
notebook==6.0.3
261
261
262
262
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
266
266
267
267
# OpenCV
268
268
RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
@@ -434,7 +434,6 @@ RUN jupyter nbextensions_configurator enable && \
434
434
jupyter lab build
435
435
436
436
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 && \
438
437
sed -i 's/source \/ usr\/ local\/ nvm\/ nvm.sh//' /etc/bash.bashrc && \
439
438
python3 -m pip uninstall -y tensorboard-plugin-wit && \
440
439
rm -rf /var/lib/apt/lists/* && \
Original file line number Diff line number Diff line change @@ -62,7 +62,15 @@ RUN python3 -m pip install --no-cache-dir \
62
62
tensorflow-plot==0.3.2 \
63
63
tensorflow_text==2.4.1 \
64
64
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
66
74
RUN python3 -m pip install --extra-index-url \
67
75
https://developer.download.nvidia.com/compute/redist nvidia-dali-cuda100 \
68
76
&& \
You can’t perform that action at this time.
0 commit comments