Merge compilation into one bash file.

Author: ruotianluo

README.md

@@ -69,41 +69,14 @@ Additional features not mentioned in the [report](https://arxiv.org/pdf/1702.021
   git clone https://github.com/ruotianluo/pytorch-faster-rcnn.git
   ```
 
-2. Choose your `-arch` option to match your GPU for step 3 and 4.
-
-  | GPU model  | Architecture |
-  | ------------- | ------------- |
-  | TitanX (Maxwell/Pascal) | sm_52 |
-  | GTX 960M | sm_50 |
-  | GTX 1080 (Ti) | sm_61 |
-  | Grid K520 (AWS g2.2xlarge) | sm_30 |
-  | Tesla K80 (AWS p2.xlarge) | sm_37 |
-
-  **Note**: You are welcome to contribute the settings on your end if you have made the code work properly on other GPUs.
-
-
-3. Build RoiPooling module
-  ```
-  cd pytorch-faster-rcnn/lib/layer_utils/roi_pooling/src/cuda
-  echo "Compiling roi_pooling kernels by nvcc..."
-  nvcc -c -o roi_pooling_kernel.cu.o roi_pooling_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_52
-  cd ../../
-  python build.py
-  cd ../../../
-  ```
-
-
-4. Build NMS
+2. Compile modules:
   ```
-  cd lib/nms/src/cuda
-  echo "Compiling nms kernels by nvcc..."
-  nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_52
-  cd ../../
-  python build.py
-  cd ../../
+  cd pytorch-faster-rcnn/lib
+  bash make.sh
+  cd ../
   ```
 
-5. Install the [Python COCO API](https://github.com/pdollar/coco). The code requires the API to access COCO dataset.
+3. Install the [Python COCO API](https://github.com/pdollar/coco). The code requires the API to access COCO dataset.
   ```Shell
   cd data
   git clone https://github.com/pdollar/coco.git

lib/make.sh

@@ -0,0 +1,22 @@
+CUDA_ARCH="-gencode arch=compute_30,code=sm_30 \
+           -gencode arch=compute_35,code=sm_35 \
+           -gencode arch=compute_50,code=sm_50 \
+           -gencode arch=compute_52,code=sm_52 \
+           -gencode arch=compute_60,code=sm_60 \
+           -gencode arch=compute_61,code=sm_61 \
+	   -gencode arch=compute_70,code=sm_70 "
+
+# Build RoiPooling module
+cd layer_utils/roi_pooling/src/cuda
+echo "Compiling roi_pooling kernels by nvcc..."
+nvcc -c -o roi_pooling_kernel.cu.o roi_pooling_kernel.cu -x cu -Xcompiler -fPIC $CUDA_ARCH 
+cd ../../
+python build.py
+cd ../../
+# Build NMS
+cd nms/src/cuda
+echo "Compiling nms kernels by nvcc..."
+nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC $CUDA_ARCH
+cd ../../
+python build.py
+cd ../