Skip to content

Commit 613760d

Browse files
authored
update readme
1 parent 813fbb0 commit 613760d

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

README.md

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,20 @@
11
# 3D Point Cloud Kernels
22
Pytorch CPU and CUDA kernels for spatial search and interpolation for 3D point clouds.
33

4-
[![PyPI version](https://badge.fury.io/py/torch-points-kernels.svg)](https://badge.fury.io/py/torch-points-kernels) ![Deploy](https://github.com/nicolas-chaulet/torch-points-kernels/workflows/Deploy/badge.svg) ![Unittests](https://github.com/nicolas-chaulet/torch-points-kernels/workflows/Unittests/badge.svg)
4+
[![PyPI version](https://badge.fury.io/py/torch-points-kernels.svg)](https://badge.fury.io/py/torch-points-kernels) [![Deploy](https://github.com/torch-points3d/torch-points-kernels/actions/workflows/deploy.yaml/badge.svg)](https://github.com/torch-points3d/torch-points-kernels/actions/workflows/deploy.yaml) [![Unittests](https://github.com/torch-points3d/torch-points-kernels/actions/workflows/tests.yaml/badge.svg)](https://github.com/torch-points3d/torch-points-kernels/actions/workflows/tests.yaml)
55

66
## Installation
7-
**Requires torch** version 1.0 or higher to be installed before proceeding. Once this is done, simply run
7+
**Update:** we now provide precompiled Conda packages for the latest PyTorch/CUDA combinations (PyTorch >= 1.10.0). To install with conda:
88
```
9-
pip install torch-points-kernels
9+
conda install -c torch-points3d torch-points-kernels
1010
```
11-
or with poetry:
11+
12+
Or, you can compile the wheel yourself for any PyTorch/CUDA combination (must have a matching installation of CUDA toolkit):
1213
```
13-
poetry add torch-points-kernels
14+
pip install torch-points-kernels
1415
```
1516

16-
To force CUDA installation (for example on Docker builds) please use the flag `FORCE_CUDA` like
17+
To force CUDA installation (for example on Docker builds) please use the flag `FORCE_CUDA`:
1718
```
1819
FORCE_CUDA=1 pip install torch-points-kernels
1920
```
Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
11
```
22
./build_conda.sh 3.9 1.9.0 cu111 # python, pytorch and cuda version
3-
```
3+
```
4+
5+
These conda scripts are based off of [pytorch_sparse](https://github.com/rusty1s/pytorch_sparse/tree/master/conda/pytorch-sparse), track changes there to update.

0 commit comments

Comments
 (0)