Commit fc56d763 authored by Daniele Venzano's avatar Daniele Venzano 🏇

Update Dockerfile to match instructions on PyTorch website

parent 36f0d438
Pipeline #5590 passed with stages
in 18 minutes and 18 seconds
......@@ -2,5 +2,5 @@ FROM jupyter/scipy-notebook:latest
MAINTAINER Daniele Venzano <venza@brownhat.org>
RUN conda install pytorch torchvision cuda80 -c soumith
RUN conda install pytorch torchvision -c pytorch
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