本人配置:
cuda10.1, gcc7.5.0, g++7.5.0
gcc --version # 查看gcc版本代码
g++ --version #查看g++版本代码
安装步骤:
pip install ninja # 安装依赖
git clone https://github.com/NVIDIA/MinkowskiEngine.git # 下载到本地
cd MinkowskiEngine # 进入目录
安装之前先把并行安装的thread数降低,否则会导致进程卡死。
进入MinkowskiEngine目录后,打开setup.py文件内位于142行左右的MAX_COMPILATION_THREADS变量值从12改成4。
Cudn10.1\10.2需要指定gcc, g++版本>=7.4.0且<8,因此个人建议7.5.0;Cudn>=11.0时,没有特定要求。
因此在cudn10.X的环境下需要执行export CXX=g++-7
,还是在MinkowskiEngine目录下执行。
# 最后安装
python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas