diff --git a/README.md b/README.md index 7320db2..50c9707 100644 --- a/README.md +++ b/README.md @@ -69,6 +69,12 @@ You may also alternatively install our library from source via: ```bash python setup.py install ``` +in the repository, or using + +``` +pip install git+https://github.com/mit-han-lab/torchsparse.git +``` +without the need to clone the repository. ## Benchmarks diff --git a/setup.py b/setup.py index fb99902..eeecf22 100644 --- a/setup.py +++ b/setup.py @@ -48,6 +48,7 @@ "torchsparse.backend", sources, extra_compile_args=extra_compile_args ) ], + url="https://github.com/mit-han-lab/torchsparse", install_requires=[ "numpy", "backports.cached_property", @@ -55,6 +56,11 @@ "typing-extensions", "wheel", "rootpath", + "torch", + "torchvision" + ], + dependency_links=[ + 'https://download.pytorch.org/whl/cu118' ], cmdclass={"build_ext": BuildExtension}, zip_safe=False,