From 2e26a2ae4935d36cc8089bf1be0072fe487b47dd Mon Sep 17 00:00:00 2001 From: Xu Ma Date: Mon, 6 Jun 2022 23:00:53 -0700 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 8e4aa12..1864000 100644 --- 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/. ```