# PyTorch version with CUDA support
PYTORCH_VERSION="2.6.0-cuda12.6-cudnn9-runtime"

# Jupyter configuration
JUPYTER_ENABLE_LAB="yes"
JUPYTER_TOKEN="pytorch"

# NVIDIA GPU configuration
NVIDIA_VISIBLE_DEVICES="all"
NVIDIA_DRIVER_CAPABILITIES="compute,utility"
GPU_COUNT=1

# Port overrides
JUPYTER_PORT_OVERRIDE=8888
TENSORBOARD_PORT_OVERRIDE=6006