diff --git a/Dockerfile b/Dockerfile index f69af25..092dc9b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,4 @@ -FROM pytorch/pytorch:1.11.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 +FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-devel RUN apt-get update && apt-get install -y libgl1-mesa-glx libpci-dev curl nano psmisc zip git && apt-get --fix-broken install -y @@ -11,4 +7,4 @@ RUN conda install -y faiss-gpu scikit-learn pandas flake8 yapf isort yacs gdown RUN pip install --upgrade pip && python -m pip install --upgrade setuptools && \ pip install opencv-python tb-nightly matplotlib logger_tt tabulate tqdm wheel mccabe scipy -COPY ./fonts/* /opt/conda/lib/python3.8/site-packages/matplotlib/mpl-data/fonts/ttf/ +COPY ./fonts/* /opt/conda/lib/python3.7/site-packages/matplotlib/mpl-data/fonts/ttf/