Update README.md

2022-06-06 23:00:53 -07:00 · 2022-06-06 23:00:53 -07:00 · 2e26a2ae49
parent 8d8e46cbf4
commit 2e26a2ae49
1 changed files with 2 additions and 0 deletions
--- 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/.
 ```