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