Add Tensorflow and PyTorch containers to the deployment process.

PawseySC / pawsey-spack-config

Automated deployment system for the scientific software stack in use at Pawsey

BSD 3-Clause "New" or "Revised" License

4 stars 9 forks source link

Add Tensorflow and PyTorch containers to the deployment process. #172

Closed dipietrantonio closed 1 year ago

dipietrantonio commented 1 year ago

The PyTorch container can be deployed with SHPC and the recent version of the associated repository. The Tensorflow container needs our custom recipe which I have developed.

dipietrantonio commented 1 year ago

Added the TensorFlow container recipe, that pulls from the more updated rocm/tensorflow DockerHub.

Commit 2c8e219