mmsegmentation框架配置

首先创建一个虚拟环境conda create -n open-mmlab python=3.7 -yconda activate open-mmlab可以参考mmsegmention的官网安装文档首先安装pytorch框架(GPU)pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://do

Shuai@

1613人浏览 · 2022-03-08 20:17:56

Shuai@ · 2022-03-08 20:17:56 发布

首先创建一个虚拟环境

conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab

可以参考mmsegmention的官网安装文档

首先安装pytorch框架(GPU)
pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

安装依赖open-mmlab
为了安装 MMCV, 我们推荐使用下面的这种预编译好的 MMCV.

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html

例如
pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html

pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html

编译mmsegmentaion

git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -e .  # 或者 "python setup.py develop"

出现的问题

显示没有发现cuda
No CUDA runtime is found, using CUDA_HOME=’/usr/local/cuda-11.5’

解决方案

天启AI社区

GitCode 天启AI是一款由 GitCode 团队打造的智能助手，基于先进的LLM（大语言模型）与多智能体 Agent 技术构建，致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话，还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力，真正做到“一句话，让 Al帮你完成复杂任务”。

更多推荐