From dc3e2761126a47e09e96e5001df0f37a87528637 Mon Sep 17 00:00:00 2001 From: Xu Ma Date: Mon, 6 Jun 2022 23:16:07 -0700 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 1864000..0413d9d 100644 --- a/README.md +++ b/README.md @@ -80,7 +80,7 @@ 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: +# if you are using Ampere GPUs (e.g., A100 and 30X0), please install compatible Pytorch and CUDA 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/.