From 271d6c5250d1ba0be59fd2f9729957dc185e8319 Mon Sep 17 00:00:00 2001 From: Xinlei Chen Date: Mon, 22 May 2017 14:41:34 -0700 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index d22b9640..0cc69659 100644 --- a/README.md +++ b/README.md @@ -55,9 +55,10 @@ Additional features not mentioned in the [report](https://arxiv.org/pdf/1702.021 2. Update your -arch in setup script to match your GPU ```Shell cd tf-faster-rcnn/lib + # Check the GPU architecture vim setup.py - # Check the GPU architecture, if you are using Pascal/Maxwell arch, sm_52 should work ``` + **Note**: I only have access to TitanX (both Maxwell and Pascal) and sm_52 seems to work fine on those machines. You are welcome to contribute the settings on your end if you have made the code work properly on other GPUs. 3. Build the Cython modules ```Shell