代码拉取完成,页面将自动刷新
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
RUN apt-get update && apt-get install g++ python3-pip python3.7-dev git -y
RUN ln -s /usr/bin/python3.7 /usr/bin/python
ENV DEBIAN_FRONTEND=noninteractive
RUN python -m pip install torch==1.4.0 torchvision==0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install tensorboardX -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install scikit-build -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN apt install cmake -y
RUN python -m pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install ipython -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install attr -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m pip install jittor -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN python -m jittor.test.test_cudnn_op
ENV DISABLE_MULTIPROCESSING=1
COPY . /workspace
WORKDIR /workspace
RUN rm /usr/bin/python3 && ln -s /usr/bin/python3.7 /usr/bin/python3
# 用来在build的时候,把模型下载到本地,因为在提交测试时,是断网测试,所以需要将模型提前下载好。
RUN python -c "from jittor.models.res2net import res2net101_26w_4s;resnet = res2net101_26w_4s(pretrained=True)"
CMD ["python", "run.py", "/input_path", "/output_path"]
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。