多余内容不做赘述,先参考这些文章进行安装,过程中遇到问题如下:
1.修改cuda版本为11.1
2.安装pytorch3.8.0+cudatoolkit11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
3.安装ME
cd MinkowskiEngine
python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas
4.安装完成之后,cuda可用,ME也正常显示版本,但是用ME.SparseTensor创建时会报错,如下:
RuntimeError: cudaErrorNoKernelImageForDevice: no kernel image is available for execution on the device
后面终于终于发现,要在ME目录下的mkfile里面修改算力,因为用到的显卡是3090,算力为8.6,
而原本文档里只支持到7.5,因此在原本基础上添加了几行代码保存重新安装即可。