Skip to content

marcj/docker-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 

Repository files navigation

Docker Alpine Pytorch v2+

Image is based on Alpine 3.9-alpine3.18 and is v2.2.1 of Pytorch.

ARM and x86_64 are available via marcjschmidt/pytorch: https://hub.docker.com/r/marcjschmidt/pytorch

Build

export VERSION=v2.2.1
docker build --build-arg VERSION=${VERSION} -t marcjschmidt/pytorch:${VERSION} .

Note: In order to get Pytorch working in Alpine, a patch is necessary. See patches/c10.patch (pytorch/pytorch#55865).

About

Pytorch v2+ in docker with ARM support

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published