diff --git a/scripts/docker/mindspore-gpu/runtime/Dockerfile b/scripts/docker/mindspore-gpu/runtime/Dockerfile index e4dc4e67dde..b32b789a579 100644 --- 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