Updating to CUDA 12.8, pytorch 2.7 and deepseepd 0.17.2

docker/variables.sh

@@ -1,9 +1,9 @@
 #!/usr/bin/env bash
 
 declare -A ALLTALK_VARS
-ALLTALK_VARS["CUDA_VERSION"]=12.6.0
+ALLTALK_VARS["CUDA_VERSION"]=12.8.1
 ALLTALK_VARS["PYTHON_VERSION"]=3.11.11
-ALLTALK_VARS["DEEPSPEED_VERSION"]=0.16.2
+ALLTALK_VARS["DEEPSPEED_VERSION"]=0.17.2
 
 # Export single variables (needed by Docker locally)
 for key in "${!ALLTALK_VARS[@]}"