Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
19 lines (12 sloc) 407 Bytes
FROM nvidia/cuda:9.0-cudnn7-runtime
RUN apt-get update && \
apt-get install -y python3.5-dev vim git g++ sudo zip python3-setuptools
RUN easy_install3 --upgrade pip setuptools
ENV PIP_CACHE_DIR=/cache PYTHONDONTWRITEBYTECODE=1
RUN pip3 install torch
RUN pip3 install torchvision
WORKDIR /app
ADD requirements.txt requirements.txt
RUN pip3 install -r requirements.txt
ADD . /app
RUN pip3 install -e .
You can’t perform that action at this time.