feat: add more

2025-10-06 21:48:39 +08:00
parent f330e00fa0
commit 3c609b5989
120 changed files with 7698 additions and 59 deletions
--- a/src/pytorch/.env.example
+++ b/src/pytorch/.env.example
@@ -0,0 +1,15 @@
+# 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