From 0b96d57f0293c925d3de5a84604ea4911f8a05e1 Mon Sep 17 00:00:00 2001 From: Xin Wang Date: Wed, 21 Oct 2020 17:11:38 -0700 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 68e22954f..fda9e923e 100755 --- a/README.md +++ b/README.md @@ -39,6 +39,8 @@ Note that you don't need to build detectron2 seperately as this codebase is self below to install the dependencies and build `FsDet`. **Requirements** +**Note**: the current codebase was tested under CUDA 10.0 and we are in the process to update the codebase to support CUDA10.2 + * Linux with Python >= 3.6 * [PyTorch](https://pytorch.org/get-started/locally/) >= 1.3 * [torchvision](https://github.com/pytorch/vision/) that matches the PyTorch installation