add mindinsight installation for dockerfile

This commit is contained in:
shenghong96 2021-08-06 15:44:36 +08:00
parent e18c277304
commit 22f702dfad
1 changed files with 4 additions and 1 deletions

View File

@ -82,4 +82,7 @@ RUN cd /tmp \
&& rm -f /tmp/openmpi-4.0.3.tar.gz
# Install MindSpore cuda-11.1 whl package
RUN pip install --no-cache-dir https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.3.0/MindSpore/gpu/x86_64/cuda-11.1/mindspore_gpu-1.3.0-cp37-cp37m-linux_x86_64.whl
RUN pip install --no-cache-dir https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.3.0/MindSpore/gpu/x86_64/cuda-11.1/mindspore_gpu-1.3.0-cp37-cp37m-linux_x86_64.whl
# Install MindInsight whl package
RUN pip install --no-cache-dir https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.3.0/MindInsight/any/mindinsight-1.3.0-py3-none-any.whl