ISBNet环境配置

最新推荐文章于 2024-09-13 19:41:31 发布

一直在路上的兔子

最新推荐文章于 2024-09-13 19:41:31 发布

阅读量302

点赞数 2

文章标签：人工智能

本文链接：https://blog.csdn.net/m0_60731542/article/details/134880528

版权

本文详细介绍了如何在Python环境中创建虚拟环境，安装特定版本的PyTorch、Spconv等库，以及如何编译和安装Segmentator、sparsehash和PointNet，最后构建并开发isbnet项目。

摘要由CSDN通过智能技术生成

1创建虚拟环境

conda create -n isbnet python=3.7
conda activate isbnet

2安装指定版本的pytorch、spconv-cu113等包

conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
pip install spconv-cu113==2.1.25
pip install torch-scatter==2.0.9 -f https://data.pyg.org/whl/torch-1.12.1+cu113.html
pip install -r requirements.txt

3 安装 Segmentator

1）下载segmentator的代码到ISBNet

git clone https://github.com/Karbo123/segmentator.git

2）编译算子

#进入文件夹

cd segmentator/csrc

#进行编译这里默认是有cmake，如果没有需要进行安装我安装的是3.28.0
mkdir build && cd build

cmake .. \
-DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'` \
-DPYTHON_INCLUDE_DIR=$(python -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") \
-DPYTHON_LIBRARY=$(python -c "import distutils.sysconfig as sysconfig; print(sysconfig.get_config_var('LIBDIR'))") \
-DCMAKE_INSTALL_PREFIX=`python -c 'from distutils.sysconfig import get_python_lib; print(get_python_lib())'`