AMD ROCm

想用 AMD 显卡跑深度学习和科学计算，却发现大多数框架只认 CUDA。ROCm 是 AMD 的开源 GPU 计算软件栈，让 PyTorch、TensorFlow 等框架在 AMD 显卡上运行。

它包含驱动、编译器、HIP 编程接口和数学库，这个仓库是 ROCm 的入口和构建说明。

软件功能

开源软件栈：驱动到库。

HIP：类 CUDA 编程。

框架支持：PyTorch、TensorFlow。