闵可夫斯基引擎Minkowski Engine安装指南

最新推荐文章于 2024-10-05 16:53:40 发布

3D感知巨头

最新推荐文章于 2024-10-05 16:53:40 发布

阅读量3.7k

点赞数 2

分类专栏：点云处理稀疏卷积文章标签：稀疏卷积 3D卷积

本文链接：https://blog.csdn.net/u013531940/article/details/127489101

版权

点云处理同时被 2 个专栏收录

1 篇文章 0 订阅

订阅专栏

稀疏卷积

1 篇文章 0 订阅

订阅专栏

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors. 有关更多信息，请访问文档页面和项目主页。

一、机器环境

操作系统为ubuntu16.04，显卡为1080Ti，ninja=1.10.2.4，anaconda环境（python=3.7，pytorch=1.8.0，torchaudio=0.8.0，torchvision=0.9.0，cudatoolkit=11.1.1）

二、检查gcc版本

# gcc -v # 如果gcc版本为5.5，一定要升级！升级gcc

$ sudo add-apt-repository ppa:ubuntu-toolchain-r/test

$ sudo apt-get update

$ sudo apt-get install gcc-7

$ sudo apt-get install g++-7

# 配置:将gcc7，g++7作为默认选项

$ sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100

$ sudo update-alternatives --config gcc

$ sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 100

$ sudo update-alternatives --config g++

三、设置环境变量

设置CUDA_HOME和LD_LIBRARY_PATH环境变量

$ export CUDA_HOME=/usr/local/cuda-11.1/
$ export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64/

设置PATH变量

# 其实就是在PATH变量中，将原先的/usr/local/cuda/bin/改成/usr/local/cuda-11.1/bin/。因为我的机器上同时存在多个cuda，默认的cuda版本是10.1，所以需要手动指定一下使用的cuda版本。设置PATH变量，是为了使用cuda-11.1中的nvcc命令。

$ export PATH=/home/autolab/bin:/home/autolab/.local/bin:/home/autolab/anaconda3/envs/SFD_PC1/bin:/home/autolab/anaconda3/condabin:/home/autolab/2021robots/tmux_depend/bin:/home/autolab/python3.8/bin:/usr/local/cuda-11.1/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin:/home/autolab/bin

设置算力变量

# 如果是最新的3090、4090等算力高的显卡，可不用设置。由于我们的显卡是1080Ti，所以需要指定一下算力。算力查询

$ export TORCH_CUDA_ARCH_LIST="6.1"

四、安装Minkowski Engine

pip install -U MinkowskiEngine --install-option="--blas=openblas" -v --no-deps

其他安装方法请移步官方的github repo。

五、算力查询

显卡计算能力
NVIDIA TITAN RTX 7.5
Geforce RTX 2080 Ti 7.5
Geforce RTX 2080 7.5
Geforce RTX 2070 7.5
Geforce RTX 2060 7.5
NVIDIA TITAN V 7.0
NVIDIA TITAN Xp 6.1
NVIDIA TITAN X 6.1
GeForce GTX 1080 Ti 6.1
GeForce GTX 1080 6.1
GeForce GTX 1070 6.1
GeForce GTX 1060 6.1
GeForce GTX 1050 6.1
GeForce GTX TITAN X 5.2
GeForce GTX TITAN Z 3.5
GeForce GTX TITAN黑色 3.5
GeForce GTX TITAN 3.5
GeForce GTX 980 Ti 5.2
GeForce GTX 980 5.2
GeForce GTX 970 5.2
GeForce GTX 960 5.2
GeForce GTX 950 5.2
GeForce GTX 780 Ti 3.5
GeForce GTX 780 3.5
GeForce GTX 770 3.0
GeForce GTX 760 3.0
GeForce GTX 750 Ti 5.0
GeForce GTX 750 5.0
GeForce GTX 690 3.0
GeForce GTX 680 3.0
GeForce GTX 670 3.0
GeForce GTX 660 Ti 3.0
GeForce GTX 660 3.0
GeForce GTX 650 Ti BOOST 3.0
GeForce GTX 650 Ti 3.0
GeForce GTX 650 3.0
GeForce GTX 560 Ti 2.1
GeForce GTX 550 Ti 2.1
GeForce GTX 460 2.1
GeForce GTS 450 2.1
GeForce GTS 450 * 2.1
GeForce GTX 590 2.0
GeForce GTX 580 2.0
GeForce GTX 570 2.0
GeForce GTX 480 2.0
GeForce GTX 470 2.0
GeForce GTX 465 2.0
GeForce GT 740 3.0
GeForce GT 730 3.5
GeForce GT 730 DDR3,128位 2.1
GeForce GT 720 3.5
GeForce GT 705 * 3.5
GeForce GT 640（GDDR5） 3.5
GeForce GT 640（GDDR3） 2.1
GeForce GT 630 2.1
GeForce GT 620 2.1
GeForce GT 610 2.1
GeForce GT 520 2.1
GeForce GT 440 2.1
GeForce GT 440 * 2.1
GeForce GT 430 2.1
GeForce GT 430 * 2.1