Update README.md

ma-xu · Jun 7, 2022 · 2e26a2a · 2e26a2a
1 parent 8d8e46c
commit 2e26a2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -80,6 +80,8 @@ conda activate pointmlp
 conda create -n pointmlp python=3.7 -y
 conda activate pointmlp
 conda install pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=10.2 -c pytorch -y
+# if you are using 3070, 3080, and 3090 GPUs, please install compatible Pytorch versions, like:
+# pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
 pip install cycler einops h5py pyyaml==5.4.1 scikit-learn==0.24.2 scipy tqdm matplotlib==3.4.2
 pip install pointnet2_ops_lib/.
 ```