The repo is obsolete. Use at your own risk. see https://github.com/pytorch/pytorch Prerequisites A ROCm enable platform. More info here. rocm-caffe2 Quick Start Install ROCm Software Stack Install Docker Build the rocm-caffe2 Project Run rocm-caffe2 Tests/Workload