CuPy has an experimental support for AMD GPU (ROCm).
- AMD GPU supported by ROCm
- ROCm: v3.5+
- See the ROCm Installation Guide for details.
The following ROCm libraries are required:
$ sudo apt install hipblas hipsparse rocsparse rocrand rocthrust rocsolver rocfft hipcub rocprim
Before installing CuPy, we recommend you to upgrade setuptools
and pip
:
$ pip install -U setuptools pip
Currently, you need to build CuPy from source to run on AMD GPU.
$ export HCC_AMDGPU_TARGET=gfx900 # This value should be changed based on your GPU $ export CUPY_INSTALL_USE_HIP=1 $ pip install cupy
Note that HCC_AMDGPU_TARGET
must be set to the ISA name supported by your GPU.
Run rocminfo
and use the value displayed in Name:
line (e.g., gfx900
).
You may also need to set ROCM_HOME
(e.g., ROCM_HOME=/opt/rocm
).