Description
Deep learning and scientific computing on AMD GPUs runs into frameworks that only know CUDA. ROCm is AMD's open source GPU computing software stack that runs PyTorch, TensorFlow and more on AMD GPUs.
It includes drivers, compilers, the HIP programming interface and math libraries; this repo is ROCm's hub and build guide.
Open stack:Drivers to libraries.
HIP:CUDA-like programming.
Frameworks:PyTorch and TensorFlow.
It includes drivers, compilers, the HIP programming interface and math libraries; this repo is ROCm's hub and build guide.
Features
Open stack:Drivers to libraries.
HIP:CUDA-like programming.
Frameworks:PyTorch and TensorFlow.
