From e8cd72a75c81fe64d5e2685897e59a69e764da5d Mon Sep 17 00:00:00 2001
From: Lucas Colas <51634013+LucasColas@users.noreply.github.com>
Date: Sun, 23 Feb 2025 11:25:54 -0500
Subject: [PATCH 1/2] New way to install with conda

---
 README.md | 49 +++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 41 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index 2f4a4ed2f..4aa90047d 100644
--- a/README.md
+++ b/README.md
@@ -40,7 +40,45 @@ We provide pretrained AdaPoinTr models (coming soon):
 | PCN |  [[Tsinghua Cloud](https://cloud.tsinghua.edu.cn/f/b822a5979762417ba75e/?dl=1)] / [[Google Drive](https://drive.google.com/file/d/17pE2U2T2k4w1KfmDbL6U-GkEwD-duTaF/view?usp=share_link)]  / [[BaiDuYun](https://pan.baidu.com/s/1KWccgcKXVIdVo4wJAmZ_8w?pwd=rc7p)](code:rc7p)  | CD = 6.53e-3|
 ## Usage
 
-### Requirements
+### With conda
+If you're on Windows, move to WSL Ubuntu.
+conda create -n my_env python=3.11
+
+conda activate my_env
+conda install nvidia/label/cuda-12.4.0::cuda-toolkit
+
+```bash
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
+```
+
+ Check that torch detects your GPU.
+```py
+import torch
+print(f"Number of GPUs available: {torch.cuda.device_count()}")
+print(f"GPU 0 name: {torch.cuda.get_device_name(0)}")
+```
+
+```bash
+pip install "git+https://github.com/facebookresearch/pytorch3d.git"
+
+pip install torch-scatter -f https://data.pyg.org/whl/torch-2.6.0+cu124.html
+
+git clone https://github.com/yuxumin/PoinTr.git
+
+cd PoinTr
+```
+Remove the == and the versions in requirements.txt
+```bash
+pip install -r requirements.txt
+
+pip install extensions/chamfer_dist extensions/cubic_feature_sampling extensions/gridding extensions/gridding_loss
+
+pip install "git+https://github.com/LucasColas/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
+
+```
+
+### Without conda
+## Requirements
 
 - PyTorch >= 1.7.0
 - python >= 3.7
@@ -55,7 +93,7 @@ We provide pretrained AdaPoinTr models (coming soon):
 pip install -r requirements.txt
 ```
 
-#### Building Pytorch Extensions for Chamfer Distance, PointNet++ and kNN
+### Building Pytorch Extensions for Chamfer Distance, PointNet++ and kNN
 
 *NOTE:* PyTorch >= 1.7 and GCC >= 4.9 are required.
 
@@ -63,13 +101,8 @@ pip install -r requirements.txt
 # Chamfer Distance
 bash install.sh
 ```
+
 The solution for a common bug in chamfer distance installation can be found in Issue [#6](https://github.com/yuxumin/PoinTr/issues/6)
-```
-# PointNet++
-pip install "git+https://github.com/erikwijmans/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
-# GPU kNN
-pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl
-```
 
 Note: If you still get `ModuleNotFoundError: No module named 'gridding'` or something similar then run these steps
 

From 7cbe5fb88319f6a1b0889784ae397d85ee358b15 Mon Sep 17 00:00:00 2001
From: Lucas Colas <51634013+LucasColas@users.noreply.github.com>
Date: Sun, 23 Feb 2025 11:28:10 -0500
Subject: [PATCH 2/2] Update README.md

---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 4aa90047d..489bad590 100644
--- a/README.md
+++ b/README.md
@@ -75,6 +75,8 @@ pip install extensions/chamfer_dist extensions/cubic_feature_sampling extensions
 
 pip install "git+https://github.com/LucasColas/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
 
+pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl
+
 ```
 
 ### Without conda