Refactoring docker images, updating CUDA to 12.6 and pytorch 2.6

This commit is contained in:
ekoc committed 2025-02-12 22:01:34 +01:00
1 parent 73d9d9450b
commit 284c100860
11 files changed
+123 -143

No files matched your search

+3 -2
View File
@@ -1,6 +1,7 @@
#!/usr/bin/env bash
export CUDA_VERSION=12.4.1
export CUDA_VERSION=12.6.0
export PYTHON_VERSION=3.11.11
export DEEPSPEED_VERSION=0.16.2
export ALLTALK_DIR=/opt/alltalk
export ALLTALK_DIR=/opt/alltalk
export GITHUB_REPOSITORY=