From eff9035e29a9cf98749505cdb9e945dbe695ffb4 Mon Sep 17 00:00:00 2001 From: mingzailao Date: Thu, 25 May 2023 13:59:23 +0800 Subject: [PATCH] no message --- .gitignore | 1 + Dockerfile | 13 +++++++++++-- 2 files changed, 12 insertions(+), 2 deletions(-) create mode 100644 .gitignore diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..f3fee9a --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +torch-whl/ diff --git a/Dockerfile b/Dockerfile index b7d349c..9085d68 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,17 @@ FROM mingzailao/cuda:11.7.0-cudnn8-devel-ubuntu18.04 +RUN rm /usr/bin/python3 +RUN ln -s /usr/bin/python3.8 /usr/bin/python3 +RUN ln -s /usr/bin/python3 /usr/bin/python +RUN apt-get install build-essential libatlas-base-dev gfortran -y +RUN pip3 install Cython RUN pip3 install --upgrade pip - -RUN pip3 install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 +COPY ./torch-whl/torch-1.13.1+cu117-cp38-cp38-linux_x86_64.whl /torch-whl/torch-1.13.1+cu117-cp38-cp38-linux_x86_64.whl +COPY ./torch-whl/torchvision-0.14.1+cu117-cp38-cp38-linux_x86_64.whl /torch-whl/torchvision-0.14.1+cu117-cp38-cp38-linux_x86_64.whl +COPY ./torch-whl/torchaudio-0.13.1+cu117-cp38-cp38-linux_x86_64.whl /torch-whl/torchaudio-0.13.1+cu117-cp38-cp38-linux_x86_64.whl +RUN pip3 install ./torch-whl/torch-1.13.1+cu117-cp38-cp38-linux_x86_64.whl +RUN pip3 install ./torch-whl/torchvision-0.14.1+cu117-cp38-cp38-linux_x86_64.whl +RUN pip3 install ./torch-whl/torchaudio-0.13.1+cu117-cp38-cp38-linux_x86_64.whl