Skip to content

Commit

Permalink
1.12.1-cuda11.3-cudnn8-devel
Browse files Browse the repository at this point in the history
Signed-off-by: siaimes <34199488+siaimes@users.noreply.github.com>
  • Loading branch information
siaimes committed Jun 23, 2023
1 parent 59a9dd6 commit 72c43c5
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -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

Expand All @@ -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/

0 comments on commit 72c43c5

Please sign in to comment.