Ubuntu系统
方法一:执行命令:
sudo apt-get install libsparsehash-dev
pip install --upgrade git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0
方法二:离线安装;首先进入该网站下载自己所需版本,mit-han-lab/torchsparse: A high-performance neural network library for point cloud processing. (github.com)
将下载好的torchsparse.tar.gz上传至项目目录中
解压:tar zxvf torchsparse版本号.tar.gz
然后安装,执行命令:pip install ./torchsparse版本号
安装失败时所遇到的坑,帮大家排雷
1.确保 PyTorch 和 NVCC 使用相同版本的 CUDA:
nvcc --version python -c "import torch; print(torch.version.cuda);"
需要确保为同一版本CUDA。
2.如果您尝试交叉编译库(即在构建时为与系统中的 GPU 不同的 GPU 进行编译,例如在 docker 构建中),请使用 TORCH_CUDA_ARCH_LIST 环境变量。您可以使用

最低0.47元/天 解锁文章
863

被折叠的 条评论
为什么被折叠?



