cupy/cupy-rocm

Sponsored OSS

By cupy

Updated 7 months ago

CuPy for AMD ROCm GPU

Image
Data Science
Languages & Frameworks
Machine Learning & AI
3

2.5K

CuPy : NumPy & SciPy for GPU

This is the image for AMD ROCm GPU platform. Refer to the Installation Guide for more details.

Docker Pull Command

docker pull cupy/cupy-rocm