modify dockerfile

2022-03-25 14:40:37 +08:00 · 2022-03-25 14:40:37 +08:00 · 8295364de0
parent 8de324cfa5
commit 8295364de0
1 changed files with 3 additions and 0 deletions
--- a/scripts/docker/mindspore-gpu/runtime/Dockerfile
+++ b/scripts/docker/mindspore-gpu/runtime/Dockerfile
@ -84,3 +84,6 @@ RUN cd /tmp \
    && make install -j4 \
    && rm -rf /tmp/openmpi-4.0.3 \
    && rm -f /tmp/openmpi-4.0.3.tar.gz
+
+# Install frequently used python packages
+RUN pip install --no-cache-dir decorator pandas sklearn pyyaml