Can you please add option for pytorch gpu alpine as well

petronetto / pytorch-alpine

A very small PyTorch container in Alpine Linux

https://hub.docker.com/r/petronetto/pytorch-alpine

BSD 3-Clause "New" or "Revised" License

66 stars 6 forks source link

Can you please add option for pytorch gpu alpine as well #5

Closed abhiksark closed 4 years ago

petronetto commented 4 years ago

I don't even know if there's support on Alpine for that. If you need to use the GPU I suggest use the official docker image, it's a way larger but it will work properly: https://hub.docker.com/r/pytorch/pytorch/tags