注意事项
- ME的github官网明确说过,pytorch使用的cuda版本必须和编译ME的cuda版本一致
- 你不需要自己安装cuda,根据pytorch官网conda安装方式会自带一个cuda运行时(但是没有nvcc等工具,但是也可以通过conda安装)
- pytorch版本一定要选择1.9.0,cuda选择11.1,最为稳妥不会出现后续乱七八糟的错误
1.安装pytorch和cuda
最好是新建一个环境。根据pytorch官网安装1.9.0 cudatoolkit=11.1版本:
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.1 -c pytorch -c conda-forge
2.安装nvcc等工具
虽然第1步已经安装了cudatoolkit,但只是一个运行时,没有编译ME所需的工具
conda install cudatoolkit-dev=11.1 -c conda-forge
检查nvcc是否正确:
which nvcc
如果输出的路径是你conda虚拟环境的路径那就继续下一步
3.安装ME
pip3 install -U MinkowskiEngine
没有报错,至此安装完成。
可能碰到的问题
- 没有安装ninja
- 没有安装openblas
这些问题可以参照这篇博客MinkowskiEngine安装总结_pycharm安装minkowskiengine-CSDN博客,或者评论区留言。