search

rocmarchive / realcaffe2

The repo is obsolete. Use at your own risk.
https://github.com/pytorch/pytorch
Apache License 2.0
12 stars 2 forks source link

add official dockerfile for AMD platform. #37

Closed petrex closed 6 years ago

petrex commented 6 years ago

Currently, the ops are validated on rocm 1.7.1.rc5

petrex commented 6 years ago

Official docker image based on rocm1.7.1 and miopen1.3 docker pull petrex/rocm_caffe2

petrex commented 6 years ago

@rohithkrn you can use this docker image in the Jenkins file as well.

dagamayank commented 6 years ago

/cc @parallelo as he is working on docker files for other frameworks as well. May be we can share the learnings and optimize from each others' efforts.

parallelo commented 6 years ago

Sounds good. I'll send you an email with a link to our (currently) internal Dockerfile setup for TF & Caffe.

petrex commented 6 years ago

WIP to adopt the modular approach (share base rocm layer with caffe and tensorflow docker)

petrex commented 6 years ago

https://github.com/pytorch/pytorch