On x86_64, cupy-11.3.0 depends on cudatoolkit-10.2, while on arm64 dependens on cudatoolkit-11.7

Description

When I install cupy on arm64: conda install cupy -c conda-forge The following packages will be downloaded:

package                    |            build
---------------------------|-----------------
**cudatoolkit-11.7.0**         |      h254b3b0_10       824.4 MB  conda-forge
cupy-11.3.0                |   py38ha9de562_1        53.4 MB  conda-forge

While on x86_64: conda install cupy -c conda-forge The following NEW packages will be INSTALLED:

cudatoolkit conda-forge/linux-64::cudatoolkit-10.2.89-h713d32c_10 None cupy conda-forge/linux-64::cupy-11.3.0-py38h69964f1_1 None

What is expected: Both install latest cuda toolkit, i.e. 11.7

To Reproduce

conda install cupy -c conda-forge

Installation

Conda-Forge (conda install ...)

Environment

# Paste the output here

Additional Information

No response

[kmaehashi]

FYI, this is the logic conda uses to detect cuda driver version in the environment: https://github.com/conda/conda/blob/main/conda/plugins/virtual_packages/cuda.py

[leofang]

I believe it means you better update your driver version in order to use cuda 11 on your x86-64 machine.