If possible, you may want to attach external storage for additional disk space. I personally use Docker for ML frameworks like PyTorch and TensorFlow - the l4t-ml container comes with both PyTorch and TensorFlow pre-installed (in addition to other ML-related packages, so it is a larger container and you may want to customize it’s Dockerfile)