https://github.com/tacc/tacc-ml
Containers for running ML applications on TACC GPU systems
https://github.com/tacc/tacc-ml
Last synced: 2 months ago
JSON representation
Containers for running ML applications on TACC GPU systems
- Host: GitHub
- URL: https://github.com/tacc/tacc-ml
- Owner: TACC
- License: bsd-3-clause
- Created: 2020-02-05T19:40:05.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2022-03-16T15:34:40.000Z (about 3 years ago)
- Last Synced: 2024-04-16T02:11:42.940Z (about 1 year ago)
- Language: Makefile
- Size: 33.2 KB
- Stars: 7
- Watchers: 9
- Forks: 3
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# TACC Machine Learning Containers
Containers for running ML applications on TACC GPU sysems
## Image information
All images are hosted on [Docker Hub](https://hub.docker.com/r/tacc/tacc-ml/tags)
Please refer to our system and version tables which support the following base operating systems:
- [Centos7](#centos7-images)
- [Ubuntu 20.04](#ubuntu2004-images)
## Usage
```
$ module load tacc-singularity
$ singularity pull docker://tacc/tacc-ml:ubuntu20.04-cuda11-tf2.6-pt1.10
$ singularity exec --nv tacc-ml_ubuntu20.04-cuda11-tf2.6-pt1.10.sif python -c 'import tensorflow as tf; print(tf.test.is_gpu_available())'
```
## Development
Begin `FROM` an image that supports your target TACC system, and add any necessary packages from there.
```
FROM tacc/tacc-ml:ubuntu20.04-cuda11-tf2.6-pt1.10
RUN conda install new_package
```
Once you are finished building your new container, push it to dockerhub and then pull to TACC.
## Centos7 Images
System
CUDA
TensorFlow
PyTorch
Maverick2
Frontera/RTX
Longhorn
Lonestar6
11
2.6
1.10
X
X
X
11
2.7
1.10
X
### centos7-cuda11-tf2.6-pt1.10
* [Dockerfile](containers/tf-conda)
* URL: `tacc/tacc-ml:centos7-cuda11-tf2.6-pt1.10`
### ppc64le-centos7-cuda11-tf2.7-pt1.10
* [Dockerfile](containers/tf-ppc64le)
* URL: `tacc/tacc-ml:ppc64le-centos7-cuda11-tf2.7-pt1.10`
## Ubuntu20.04 Images
System
CUDA
TensorFlow
PyTorch
Maverick2
Frontera/RTX
Longhorn
Lonestar6
11
2.6
1.10
X
X
X
11
2.7
1.10
X
### ubuntu20.04-cuda11-tf2.6-pt1.10
* [Dockerfile](containers/tf-conda)
* URL: `tacc/tacc-ml:ubuntu20.04-cuda11-tf2.6-pt1.10`
### ppc64le-ubuntu20.04-cuda11-tf2.7-pt1.10
* [Dockerfile](containers/tf-ppc64le)
* URL: `tacc/tacc-ml:ppc64le-ubuntu20.04-cuda11-tf2.7-pt1.10101010101010101010`