Docker Hub
hub.docker.com › searchPyTorch is a deep learning framework that puts Python first. Container. Linux. x86-64. 1M+ Downloads. 12 Stars. pytorchlightning/pytorch_lightning
Docker Hub
hub.docker.com › r › pytorchPyTorch is a deep learning framework that puts Python first. Container. Pulls 5M+ Overview Tags. PyTorch is a deep learning framework that puts Python first. It provides Tensors a
serve/README.md at master · pytorch/serve · GitHub
https://github.com/pytorch/serve/blob/master/docker/README.md09.11.2021 · -g, --gpu: Build image with GPU based ubuntu base image-bt, --buildtype: Which type of docker image to build. Can be one of : production, dev, codebuild-t, --tag: Tag name for image. If not specified, script uses torchserve default tag names.-cv, --cudaversion: Specify to cuda version to use. Supported values cu92, cu101, cu102, cu111, cu113 ...