Skip to content

Artrajz/pytorch-docker

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

构建一个用于推理的torch环境,作为基础镜像在Dockerfile中使用,

CPU 支持amd64和arm64

FROM artrajz/pytorch:1.13.1-cpu-py3.10.11-ubuntu22.04

GPU 仅支持amd64

FROM artrajz/pytorch:1.13.1-cu117-py3.10.11-ubuntu22.04

About

Basic Docker image for torch used for inference.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published