mmcv with cuda 12.1

[isobelm]

Is it possible to use mmcv with cuda 12.1?

I am using Ubuntu 22.04.2 I have python 3.8.16, pytorch 2.1.0 and openmim 0.3.9.
I have been trying to use mim install "mmcv>=2.0.0" but it always ends with:

  ERROR: Failed building wheel for mmcv
  Running setup.py clean for mmcv
Failed to build mmcv
ERROR: Could not build wheels for mmcv, which is required to install pyproject.toml-based projects

[FORRYKIM]

have u sloved yet?

[samskalicky]

Hi guys, you can build mmcv with CUDA 12 by following the instructions to build from source. heres a couple pointers:
https://github.com/open-mmlab/mmcv/?tab=readme-ov-file#install-mmcv
https://mmcv.readthedocs.io/en/latest/get_started/build.html

mmcv/docker/dev/Dockerfile

Lines 28 to 31 in 761d780

	RUN git clone https://github.com/open-mmlab/mmcv.git /mmcv
	WORKDIR /mmcv
	RUN git checkout 2.x && git rev-parse --short HEAD
	RUN pip install --no-cache-dir -e .[all] -v && pip install pre-commit && pre-commit install

mmcv/setup.py

Lines 159 to 170 in 761d780

	cuda_args = os.getenv('MMCV_CUDA_ARGS')
	extra_compile_args = {
	'nvcc': [cuda_args, '-std=c++14'] if cuda_args else ['-std=c++14'],
	'cxx': ['-std=c++14'],
	}
	if torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':
	define_macros += [('MMCV_WITH_CUDA', None)]
	extra_compile_args['nvcc'] += [
	'-D__CUDA_NO_HALF_OPERATORS__',
	'-D__CUDA_NO_HALF_CONVERSIONS__',
	'-D__CUDA_NO_HALF2_OPERATORS__',
	]

heres some reference steps to customize the CUDA build (ie. to build for the latest Nvidia H200 GPU with CUDA 9.0 arch)

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
export FORCE_CUDA=1
export MMCV_CUDA_ARGS="-gencode arch=compute_90,code=compute_90
pip install -e .