use FA 2.8.3 which is compatible with torch 2.8.0 (#3936)

* use FA 2.8.3

* Disable fail-fast in Docker publish workflow

* fix

* change apt source

Author: lvhan028

.github/workflows/docker.yml

@@ -36,6 +36,7 @@ jobs:
     runs-on: ubuntu-latest
     environment: 'prod'
     strategy:
+      fail-fast: false
       matrix:
         cuda_version: ['cu12.8', cu12, cu11]
     env:
@@ -80,7 +81,7 @@ jobs:
           echo $TAG
           docker build . -f docker/Dockerfile -t ${TAG} --build-arg CUDA_VERSION=${CUDA_VERSION}
           docker push $TAG
-      - name: Push Docker image latest-cu12 as latest
+      - name: Push Docker image as latest
         if: endsWith(env.TAG, 'latest-cu12') == true
         run: |
           export latest_TAG=${TAG_PREFIX}:latest

docker/Dockerfile

@@ -21,6 +21,7 @@ ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=Etc/UTC
 
 RUN --mount=type=cache,target=/root/.cache \
+    sed -i 's|http://archive.ubuntu.com|http://azure.archive.ubuntu.com|g' /etc/apt/sources.list && \
     apt-get update -y && \
     apt-get install -y --no-install-recommends \
         tzdata wget curl openssh-server ssh sudo git-core \

docker/install.sh

@@ -1,6 +1,7 @@
 #!/bin/bash -ex
 
 export DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC
+sed -i 's|http://archive.ubuntu.com|http://azure.archive.ubuntu.com|g' /etc/apt/sources.list
 apt-get update -y
 apt-get install -y --no-install-recommends \
     tzdata wget curl ssh sudo git-core libibverbs1 ibverbs-providers ibverbs-utils librdmacm1 libibverbs-dev rdma-core libmlx5-1
@@ -35,7 +36,7 @@ if [[ "${CUDA_VERSION_SHORT}" = "cu118" ]]; then
     FA_VERSION=2.7.3
     TORCH_VERSION="<2.7"
 else
-    FA_VERSION=2.8.2
+    FA_VERSION=2.8.3
     TORCH_VERSION=""
 fi