From bc5d4800c5316562380c81f1d52df8e275738345 Mon Sep 17 00:00:00 2001 From: Alexander Penkin Date: Sat, 30 Nov 2019 11:24:00 +0500 Subject: [PATCH] Add PyTorch MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Команда установки из https://pytorch.org/ --- clients/python/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/clients/python/Dockerfile b/clients/python/Dockerfile index 430e35b..d239999 100644 --- a/clients/python/Dockerfile +++ b/clients/python/Dockerfile @@ -1,6 +1,7 @@ FROM python RUN pip install numpy cython +RUN pip install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html COPY . /project -WORKDIR /project \ No newline at end of file +WORKDIR /project