Refactoring docker images, updating CUDA to 12.6 and pytorch 2.6
This commit is contained in:
1 parent
73d9d9450b
commit
284c100860
11 files changed
+123
-143
No files matched your search
+3
-2
@@ -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=
|
||||
Reference in new issue
Block a user