Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

add the option to use a FORCE_CUDA to force cuda installation on docker #612

Merged
merged 12 commits into from
Mar 31, 2019
Merged

add the option to use a FORCE_CUDA to force cuda installation on docker #612

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
91b36ee
add a FORCE_CUDA flag
obendidi Mar 27, 2019
bdb36fa
Update Dockerfile
obendidi Mar 27, 2019
da40de3
Update setup.py
obendidi Mar 27, 2019
657e962
add FORCE_CUDA as an ARG
Mar 28, 2019
41a0643
modified: docker/Dockerfile
obendidi Mar 28, 2019
7126480
small fix to readme of demo
Mar 28, 2019
8f1b4a2
remove test print
Mar 28, 2019
460d059
keep ARG_CUDA
Mar 28, 2019
52d7ccf
remove env value and use the one from ARG
Mar 28, 2019
ae6ede3
keep same formatting as source
obendidi Mar 28, 2019
5afb532
change proposed by @miguelvr
Mar 28, 2019
e676c95
Update INSTALL.md
obendidi Mar 28, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@ RUN git clone https://github.com/cocodataset/cocoapi.git \
&& python setup.py build_ext install

# install PyTorch Detection
RUN git clone https://github.com/bendidi/maskrcnn-benchmark.git \
RUN git clone https://github.com/facebookresearch/maskrcnn-benchmark.git \
&& cd maskrcnn-benchmark \
&& FORCE_CUDA=1 python setup.py build develop
obendidi marked this conversation as resolved.
Show resolved Hide resolved
&& python setup.py build develop

WORKDIR /maskrcnn-benchmark