update

zju3dv · Jun 23, 2019 · 460827f · 460827f
1 parent a85723f
commit 460827f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -21,7 +21,9 @@ Check [TRUNCATION_LINEMOD.md](TRUNCATION_LINEMOD.md) for information about the T
 pip install -r requirements.txt
 ```
 
-We need compile several files, which works fine with pytorch v0.4.1 and gcc 5.4.0.
+We need compile several files, which works fine with pytorch v0.4.1/v1.0.1 and gcc 5.4.0.
+
+For users with RTX GPU, you must use CUDA10 and pytorch v1.0.1 built from CUDA10.
 
 2. Compile the Ransac Voting Layer