16 lines
348 B
Plaintext
16 lines
348 B
Plaintext
# 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
|