Sign inSign up

uvarc/pytorch

By uvarc

•Updated about 3 years ago

Image
0

2.5K

uvarc/pytorch repository overview

⁠PyTorch

https://pytorch.org/⁠

⁠Tags

  • x.y.z contains additional preinstalled Python packages (see below). This is the recommended tag for most Rivanna users.
  • x.y.z-<user> is a custom container prepared for a specific user.

All are GPU-compatible.

⁠Usage on Rivanna

module load singularity
singularity pull docker://uvarc/pytorch:x.y.z
singularity run --nv pytorch_x.y.z.sif your_script.py

⁠Preinstalled Python Packages

Package\PyTorch1.12.01.10.01.8.0/11.7.01.6.0
Python3.9.133.8.123.7.3/3.8.83.7.33.7.3
NumPy1.23.11.20.11.20.11.19.21.19.1
SciPy1.8.11.6.11.6.11.5.31.5.2
TorchVision0.13.00.11.10.9.0/10.8.10.7.0
Torchaudio0.12.00.10.00.8.0/10.7.00.6.0
PyTorch Lightning1.6.5+1.5.3+1.2.1+1.0.30.8.5
TorchText0.13.00.11.00.9.0/10.8.00.7.0
BoTorch0.6.50.5.10.4.00.3.20.3.0
Ignite1.1.01.1.01.1.01.1.00.4.1
LightGBM2.3.1
Matplotlib3.5.23.5.03.3.43.3.23.3.0
Seaborn0.11.20.11.20.11.10.11.00.10.1
Pandas1.4.31.3.41.2.31.1.31.1.0
Scikit-learn1.1.11.0.10.24.10.23.20.23.2
Scikit-image0.19.30.18.30.18.10.17.20.17.2
OpenSlide-Python1.2.01.1.21.1.21.1.21.1.1
OpenCV4.6.0.664.5.4.604.5.1.484.4.0.444.3.0.36

To install more packages:

/path/to/sif -m pip install --user <package>

Tag summary

Content type

Image

Digest

sha256:7ff1ea3f6…

Size

9.1 GB

Last updated

about 3 years ago

docker pull uvarc/pytorch:2.0.1