GitHub Workflow安装MMCV时遇CUDA_HOME未设置错误求助

1. 修正PyTorch版本至预期版本 #

原安装命令未指定版本，默认拉取了最新的2.0.0版本。需明确指定目标版本，以适配后续MM系列框架的版本要求：

• 修改PyTorch安装命令，例如预期安装1.13.1版本（需根据实际需求调整版本号）：

  pip3 install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cu117
  

• 安装完成后可通过python -c "import torch; print(torch.__version__)"验证版本是否符合预期。

2. 解决CUDA_HOME未设置的报错 #

MMCV等包在编译时需要依赖CUDA环境，GitHub Workflow默认可能未配置CUDA_HOME变量，可通过以下两种方式处理：

方式一：手动设置CUDA_HOME环境变量 #

在Workflow的步骤中添加环境变量配置：

- name: Configure CUDA_HOME
  run: |
    # 若使用GitHub官方提供的带CUDA的runner，CUDA通常安装在/usr/local/cuda-11.7路径
    echo "CUDA_HOME=/usr/local/cuda-11.7" >> $GITHUB_ENV
    echo "PATH=$CUDA_HOME/bin:$PATH" >> $GITHUB_ENV
    echo "LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH" >> $GITHUB_ENV

方式二：使用预编译的MM系列包（推荐） #

避免本地编译，直接安装适配cu117和对应PyTorch版本的预编译轮子，无需依赖CUDA_HOME：

• 先安装mmengine：

  pip3 install mmengine==0.6.0
  

• 再安装适配cu117和目标PyTorch版本的mmcv（以PyTorch1.13为例）：

  pip3 install mmcv==2.0.0rc3 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html
  

• 最后安装mmdet和mmaction2：

  pip3 install mmdet==3.0.0rc5 mmaction2==1.0rc3
  

3. 版本兼容性验证 #

确保所有组件版本匹配：

• 确认MMDet、MMAction2的版本与MMCV、PyTorch版本兼容（例如MMDet3.0.0rc5适配MMCV2.0.0rc3，且支持PyTorch1.13~2.0）
• 若仍有问题，可在Workflow中添加步骤，输出当前环境的CUDA版本、PyTorch版本，便于排查：

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

内容的提问来源于stack exchange，提问作者ThomasEdwin