RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine

创建conda环境

conda create -n mink python=3.8
source activate mink

安装torch1.9.0 + torchvision

可到此处下载whl文件(也可以直接pip安装)

pip install torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl

然后安装torchvision

pip install torchvision

安装完后要看torch对应的cuda版本是否为cuda11.1

python -c "import torch; print(torch.version.cuda)"

安装open3d

pip install open3d

安装MinkowskiEngine

下载MinkowskiEngine代码库并安装

git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine
python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas

可能出现的错误

fatal error: cblas.h: No such file or directory

解决办法

sudo apt-get install libopenblas-dev

你可能感兴趣的:(深度学习,pytorch,3D图形学)