pytorch下载
根据个人CUDA版本下载相应的pytorch(我用的cuda11.1,pytorch1.9.0):
https://pytorch.org/get-started/previous-versions/
torch-cluster和torch-scatter下载
torch-cluster和torch-scatter要下载与pytorch对应版本的:
https://data.pyg.org/whl/
spconv
spconv 1.1下载:
https://github.com/traveller59/spconv/tree/v1.1
第三方库pybind 11下载:
https://github.com/pybind/pybind11
下载解压后,将pybind11里的内容复制到spconv项目中的third_party对应的目录中。
执行spconv目录中的setup.py:
python setup.py bdist_wheel
上一步完成后,spconv目录下会多出几个文件,先进入到dist目录下,然后安装该目录下的whl文件
cd ./dist
pip install spconv-1.0-cp36-cp36m-linux_x86_64.whl
查看结果
conda list | grep spconv