Browse Source

[to #44790143]fix: add ipythonkernel to image for dsw

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10049527

    * add ipykernel to image for dsw
master
mulin.lyh 3 years ago
parent
commit
ad6bb1e7d9
2 changed files with 4 additions and 2 deletions
  1. +3
    -1
      docker/Dockerfile.ubuntu
  2. +1
    -1
      tests/run.py

+ 3
- 1
docker/Dockerfile.ubuntu View File

@@ -75,7 +75,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
ENV SHELL=/bin/bash ENV SHELL=/bin/bash


# install special package # install special package
RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 numpy==1.18.5 datasets==2.1.0
RUN pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 datasets==2.1.0 ipykernel && \
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
pip config set install.trusted-host pypi.tuna.tsinghua.edu.cn


RUN if [ "$USE_GPU" = "True" ] ; then \ RUN if [ "$USE_GPU" = "True" ] ; then \
pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \ pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \


+ 1
- 1
tests/run.py View File

@@ -420,7 +420,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--suites', '--suites',
nargs='*', nargs='*',
help='Run specified test suites(test suite file list)')
help='Run specified test suites(test suite files list split by space)')
args = parser.parse_args() args = parser.parse_args()
set_test_level(args.level) set_test_level(args.level)
os.environ['REGRESSION_BASELINE'] = '1' os.environ['REGRESSION_BASELINE'] = '1'


Loading…
Cancel
Save