From 53cdf75b23040c5b4f8e1e3d8d9c32403b06dcb7 Mon Sep 17 00:00:00 2001 From: mingzailao Date: Sun, 19 Nov 2023 15:15:07 +0800 Subject: [PATCH] no message --- Dockerfile | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/Dockerfile b/Dockerfile index 1fab714..cadf683 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,15 +1,17 @@ -FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel - -# encoding -RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment -RUN echo "en_US.UTF-8 UTF-8" >> /etc/locale.gen -RUN echo "LANG=en_US.UTF-8" > /etc/locale.conf +FROM mingzailao/cuda:10.1-cudnn7-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 -# fix (tzdata) -ARG DEBIAN_FRONTEND=noninteractive - +COPY ./torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl /torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl +COPY ./torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl /torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl +RUN pip3 install ./torch-whl/torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl +RUN pip3 install ./torch-whl/torchvision-0.8.0-cp38-cp38-linux_x86_64.whl