mmcv安装教程 cuda12.4

安装MMDetection（mmcv）库支持CUDA 12.4，你需要遵循以下步骤：

1. **环境准备**：
- 首先，确保已安装Python（建议使用Python 3.6以上版本）以及pip。
- 安装PyTorch，推荐选择与CUDA 12.4兼容的版本，例如PyTorch 1.8.0或更高。你可以从[这里](https://pytorch.org/get-started/locally/)下载并按照说明安装。

2. **安装mmcv**：
使用命令行，运行以下命令来安装mmcv及其依赖项，记得指定`--cuda-root`指向CUDA 12.4的安装路径：

   pip install mmcv-full==x.y.z --extra-index-url https://download.openmmlab.com/mmcv/dist/cu124/torch1.8/index.html --cuda-root=/path/to/your/cuda-12.4

请将`x.y.z`替换为mmcv的最新稳定版本号，可以在[MMDetection官网](https://github.com/open-mmlab/mmdetection)或通过`pip list | grep mmcv-full`查看当前可用版本。

3. **配置环境变量**：
确保CUDA、cuDNN和PATH设置正确。编辑`~/.bashrc`或其他适当的启动文件，添加类似下面的内容：

   export CUDA_HOME=/path/to/your/cuda-12.4
   export PATH=$CUDA_HOME/bin:$PATH
   export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
   source ~/.bashrc

4. **测试安装**：
安装完成后，打开一个新的终端会话，并尝试运行一些基本的mmcv示例或MMDetection项目来验证安装是否成功。