From 3041486e6b2f60effa90333c02b05e4002a63538 Mon Sep 17 00:00:00 2001 From: Dan Sun Date: Sun, 10 Nov 2019 12:41:39 -0500 Subject: [PATCH] Install pytorch-cpu for cpu image (#536) --- python/pytorch.Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/pytorch.Dockerfile b/python/pytorch.Dockerfile index 2db7c2e95c0..4a8c1bf69bc 100644 --- a/python/pytorch.Dockerfile +++ b/python/pytorch.Dockerfile @@ -17,7 +17,7 @@ RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-4. /opt/conda/bin/conda create -y --name pytorch-py37 python=3.7.3 numpy pyyaml scipy ipython mkl&& \ /opt/conda/bin/conda clean -ya ENV PATH /opt/conda/envs/pytorch-py37/bin:$PATH -RUN conda install --name pytorch-py37 pytorch torchvision -c soumith && /opt/conda/bin/conda clean -ya +RUN conda install --name pytorch-py37 pytorch-cpu torchvision -c soumith && /opt/conda/bin/conda clean -ya WORKDIR /workspace RUN chmod -R a+w /workspace @@ -25,4 +25,4 @@ RUN chmod -R a+w /workspace COPY . . RUN pip install --upgrade pip && pip install -e ./kfserving RUN pip install -e ./pytorchserver -ENTRYPOINT ["python", "-m", "pytorchserver"] \ No newline at end of file +ENTRYPOINT ["python", "-m", "pytorchserver"]