Commit 54f9e98c authored by Daniele Venzano's avatar Daniele Venzano

Install pytorch with pip and the jupyter image

parent d27e2176
Pipeline #5622 passed with stages
in 25 minutes and 30 seconds
FROM gcr.io/tensorflow/tensorflow:1.3.0-gpu-py3
FROM jupyter/datascience-notebook:latest
MAINTAINER Daniele Venzano <venza@brownhat.org>
RUN pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp35-cp35m-linux_x86_64.whl
RUN pip3 install torchvision
RUN pip install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
RUN pip install torchvision
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment