-
Notifications
You must be signed in to change notification settings - Fork 7
/
Dockerfile
27 lines (14 loc) · 1.32 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel AS builder
RUN apt-key del 7fa2af80 && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
WORKDIR /
RUN apt-get update && apt-get install -y curl zip
RUN curl https://dl.freefontsfamily.com/download/Times-New-Roman-Font/ -o Times-New-roman.zip
RUN unzip Times-New-roman.zip
RUN mkdir Times-New-Roman
RUN cp "Times New Roman"/* "Times-New-Roman/"
FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
RUN apt-key del 7fa2af80 && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN apt-get update && apt-get --fix-broken install -y && apt-get install -y libgl1-mesa-glx libpci-dev curl nano psmisc zip git
RUN conda install -y faiss-gpu scikit-learn pandas flake8 yapf isort yacs gdown future libgcc -c conda-forge
RUN pip install opencv-python tb-nightly matplotlib logger_tt tabulate tqdm scipy
COPY --from=builder /Times-New-Roman/* /opt/conda/lib/python3.7/site-packages/matplotlib/mpl-data/fonts/ttf/