Skip to content

benborder/libtorch-docker

Repository files navigation

libtorch-docker

Creates libtorch based docker images for both cpu and cuda

Building locally

docker build -f cpu.Dockerfile -t libtorch-docker:cpu-dev .
docker build -f cuda.Dockerfile -t libtorch-docker:cuda-dev .

About

Provides libtorch docker images

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages