Sign inSign up

sanglu/torch

By sanglu

•Updated over 1 year ago

This project sets up a Docker environment for PyTorch with CUDA 12.1.1 support.

Image
Machine learning & AI
Developer tools
0

237

Tags for sanglu/torch

Sort by

TAG

cuda-12.8.1

Last pushed over 1 year by sanglu

docker pull sanglu/torch:cuda-12.8.1
DigestOS/ARCHCompressed size

39c34baef39d

linux/amd64

12.25 GB

TAG

cuda-12.4.1

Last pushed over 1 year by sanglu

docker pull sanglu/torch:cuda-12.4.1
DigestOS/ARCHCompressed size

ae4e7315f40d

linux/amd64

9.54 GB

TAG

cuda-12.1.1

Last pushed over 1 year by sanglu

docker pull sanglu/torch:cuda-12.1.1
DigestOS/ARCHCompressed size

c42f887ea628

linux/amd64

9.12 GB

TAG

cuda-11.8.0

Last pushed over 1 year by sanglu

docker pull sanglu/torch:cuda-11.8.0
DigestOS/ARCHCompressed size

2ebb40e5e01b

linux/amd64

9.31 GB