cupy/cupy-rocm
Sponsored OSS
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