mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
[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
This commit is contained in:
@@ -75,7 +75,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
|
||||
ENV SHELL=/bin/bash
|
||||
|
||||
# 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 \
|
||||
pip install --no-cache-dir dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html; \
|
||||
|
||||
@@ -420,7 +420,7 @@ if __name__ == '__main__':
|
||||
parser.add_argument(
|
||||
'--suites',
|
||||
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()
|
||||
set_test_level(args.level)
|
||||
os.environ['REGRESSION_BASELINE'] = '1'
|
||||
|
||||
Reference in New Issue
Block a user