[pytorch] [build] [training] [ec2, sagemaker] Update accelerate version to 1.8.1 (#4947)

jinyan-li1 · web-flow · commit db72100f0b4c · 2025-06-23T08:51:36.000-07:00
update accelerate version to 1.8.1 for PT 2.6
diff --git a/pytorch/training/docker/2.6/py3/Dockerfile.ec2.cpu.core_packages.json b/pytorch/training/docker/2.6/py3/Dockerfile.ec2.cpu.core_packages.json
@@ -1,6 +1,6 @@
 {
   "accelerate": {
-    "version_specifier": "==1.8.0",
+    "version_specifier": "==1.8.1",
     "skip": "True"
   },
   "s3torchconnector": {
diff --git a/pytorch/training/docker/2.6/py3/Dockerfile.sagemaker.cpu.core_packages.json b/pytorch/training/docker/2.6/py3/Dockerfile.sagemaker.cpu.core_packages.json
@@ -1,6 +1,6 @@
 {
   "accelerate": {
-    "version_specifier": "==1.8.0",
+    "version_specifier": "==1.8.1",
     "skip": "True"
   },
   "s3torchconnector": {
diff --git a/pytorch/training/docker/2.6/py3/cu126/Dockerfile.ec2.gpu.core_packages.json b/pytorch/training/docker/2.6/py3/cu126/Dockerfile.ec2.gpu.core_packages.json
@@ -1,6 +1,6 @@
 {
   "accelerate": {
-    "version_specifier": "==1.8.0",
+    "version_specifier": "==1.8.1",
     "skip": "True"
   },
   "flash-attn": {
diff --git a/pytorch/training/docker/2.6/py3/cu126/Dockerfile.sagemaker.gpu.core_packages.json b/pytorch/training/docker/2.6/py3/cu126/Dockerfile.sagemaker.gpu.core_packages.json
@@ -1,6 +1,6 @@
 {
   "accelerate": {
-    "version_specifier": "==1.8.0",
+    "version_specifier": "==1.8.1",
     "skip": "True"
   },
   "flash-attn": {
