Skip to content

Instantly share code, notes, and snippets.

@curtkim

curtkim/Dockerfile

Created July 19, 2020 23:45
Show Gist options
  • Save curtkim/b467a9fdf1ab06f327fe907891a4124b to your computer and use it in GitHub Desktop.
Save curtkim/b467a9fdf1ab06f327fe907891a4124b to your computer and use it in GitHub Desktop.
Download ZIP
stereo-rcnn
Raw
Dockerfile
FROM tiangolo/python-machine-learning:cuda9.1-devel-python3.6
RUN conda install pytorch=1.0.0 cuda90 -c pytorch
RUN conda install torchvision -c pytorch
WORKDIR /app
COPY . /app
RUN pip install -r requirements.txt
RUN cd lib && python setup.py build develop
RUN cd /app
# docker run -v --gpus all -it -v $(pwd):/data stereo-rcnn bash
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment