From 8da35e03560b8ae689a6dc9f4d99ef7d5425f2c0 Mon Sep 17 00:00:00 2001
From: Yingqi Cao <78573313+ioeddk@users.noreply.github.com>
Date: Tue, 30 Jul 2024 21:09:25 -0700
Subject: [PATCH] Installation from git directly.  (#319)

* Update setup.py

added url to setup.py for PyPI Server test.

* Update setup.py

* Update setup.py

* Update setup.py

Updated the url to be the public repo

* Update README.md
---
 README.md | 6 ++++++
 setup.py  | 6 ++++++
 2 files changed, 12 insertions(+)

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,