RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine

最新推荐文章于 2024-02-13 11:26:26 发布

转载最新推荐文章于 2024-02-13 11:26:26 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://blog.csdn.net/weixin_43486780/article/details/119849000

文章标签：

#python #pytorch #深度学习

该博客详细介绍了如何在CUDA11.1环境下，针对RTX3090显卡安装torch1.9.0和torchvision，接着演示了安装open3d的步骤，并重点讲解了安装MinkowskiEngine过程中可能遇到的`cblas.h`缺失问题及其解决方案，包括安装libopenblas-dev。整个过程覆盖了conda环境创建、whl文件下载和Python库的安装。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine

转载自 https://blog.csdn.net/weixin_43486780/article/details/119849000

创建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

可能出现的错误