Update INSTALL.md

INSTALL.md

@@ -26,8 +26,8 @@ conda install ipython
 pip install ninja yacs cython matplotlib tqdm
 
 # follow PyTorch installation in https://pytorch.org/get-started/locally/
-# we give the instructions for CUDA 9.0
-conda install -c pytorch torchvision=0.2.1 cudatoolkit=9.0
+# we give the instructions for CUDA 10.2
+conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
 
 export INSTALL_DIR=$PWD