yolo-world环境配置

学渣676562025-07-16 14:56

CUDA 11.6

conda install pytorch1.13.1 torchvision0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

bash 复制代码

# 安装虚拟环境
# 1. 创建 conda 环境
conda create -n yolo-world python=3.9
conda activate yolo-world

# 2. 安装 PyTorch + CUDA 11.6（使用 conda）
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

# 如果 conda 下载慢，可以改用 pip（但需先安装 conda 的 cudatoolkit）：
# conda install cudatoolkit=11.6 -c nvidia
# pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple

# 3. 安装其他依赖（用 pip 并指定清华源）
pip install requests==2.28.2 tqdm==4.65.0 rich==13.4.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -U openmim -i https://pypi.tuna.tsinghua.edu.cn/simple

# 4. 通过 mim 安装 MM 系列库
mim install mmcv==2.1.0
mim install mmdet==3.3.0
mim install mmengine==0.10.4
mim install mmyolo==0.6.0

# 注意：mmcv-lite 和 mmcv 冲突，二选一即可
# mim install mmcv-lite==2.0.1  # 如果不需要完整版 mmcv

抱抱脸国内连接不上

bash 复制代码

git config --global url."https://hf-mirror.com".insteadOf "https://huggingface.co"
git clone https://hf-mirror.com/openai/clip-vit-base-patch32