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
mit-han-lab · Jul 31, 2024 · 8da35e0 · 8da35e0
1 parent ab49a6e
commit 8da35e0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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
@@ -48,13 +48,19 @@
             "torchsparse.backend", sources, extra_compile_args=extra_compile_args
         )
     ],
+    url="https://github.com/mit-han-lab/torchsparse",
     install_requires=[
         "numpy",
         "backports.cached_property",
         "tqdm",
         "typing-extensions",
         "wheel",
         "rootpath",
+        "torch",
+        "torchvision"
+    ],
+    dependency_links=[
+        'https://download.pytorch.org/whl/cu118'
     ],
     cmdclass={"build_ext": BuildExtension},
     zip_safe=False,