search

floydhub / dockerfiles

Deep Learning Dockerfiles
https://docs.floydhub.com/guides/environments/
Apache License 2.0
156 stars 57 forks source link

Suggestion for adding Magma to Pytorch 1.0.0 image #61

Closed wangell closed 6 years ago

wangell commented 6 years ago

I need MAGMA support in Pytorch for gpytorch, would greatly appreciate it if your Pytorch build supported it so I can continue using floydhub.

I tried to add a method to your current docker build system so you could integrate it into other images over time.

ReDeiPirati commented 6 years ago

Hi @wangell,

Thanks for the PR, could you also add a test for this in the pytorch test folder, please?

wangell commented 6 years ago

Added a small minimal test using symeig

houqp commented 6 years ago

Thanks @wangell ! Will kick off our automated build system now. It will take couple days before everything is built, tested and gradually rolled out to all our machines so we can closely monitor any breaking changes.