移除cu121增加cu126 #76

.github/workflows/gpu.yml

@@ -39,63 +39,63 @@ jobs:
           token: ${{ secrets.MY_GITHUB_TOKEN }}
           omitBody: true
 
-  release_linux_cu121_glibc232:
+  release_linux_cu124_glibc232:
     runs-on: ubuntu-22.04
     steps:
       - uses: actions/checkout@v4
       - name: setup cuda
         uses: Jimver/[email protected]
         with:
-            cuda: '12.1.0'
+            cuda: '12.4.0'
             method: network
 
       - name: make
         working-directory: ./lib
         run: |
           sudo apt-get clean
           sudo apt-get install -y cmake
-          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu121.zip
+          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu124.zip
           unzip -qq libtorch.zip
           export TORCH_CUDA_ARCH_LIST="5.0 8.0 8.6 8.9 9.0"
           cmake -DCMAKE_PREFIX_PATH=./libtorch -DCMAKE_BUILD_TYPE=Release .
           make -j
-          mv libgotorch.so libgotorch_gpu.cu121.glibc232.so
+          mv libgotorch.so libgotorch_gpu.cu124.glibc232.so
 
       - name: upload
         uses: ncipollo/release-action@v1
         with:
-          artifacts: ./lib/libgotorch_gpu.cu121.glibc232.so
+          artifacts: ./lib/libgotorch_gpu.cu124.glibc232.so
           allowUpdates: true
           generateReleaseNotes: true
           token: ${{ secrets.MY_GITHUB_TOKEN }}
           omitBody: true
 
-  release_linux_cu124_glibc232:
+  release_linux_cu126_glibc232:
     runs-on: ubuntu-22.04
     steps:
       - uses: actions/checkout@v4
       - name: setup cuda
         uses: Jimver/[email protected]
         with:
-            cuda: '12.4.0'
+            cuda: '12.6.0'
             method: network
 
       - name: make
         working-directory: ./lib
         run: |
           sudo apt-get clean
           sudo apt-get install -y cmake
-          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu124.zip
+          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu126/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu126.zip
           unzip -qq libtorch.zip
           export TORCH_CUDA_ARCH_LIST="5.0 8.0 8.6 8.9 9.0"
           cmake -DCMAKE_PREFIX_PATH=./libtorch -DCMAKE_BUILD_TYPE=Release .
           make -j
-          mv libgotorch.so libgotorch_gpu.cu124.glibc232.so
+          mv libgotorch.so libgotorch_gpu.cu126.glibc232.so
 
       - name: upload
         uses: ncipollo/release-action@v1
         with:
-          artifacts: ./lib/libgotorch_gpu.cu124.glibc232.so
+          artifacts: ./lib/libgotorch_gpu.cu126.glibc232.so
           allowUpdates: true
           generateReleaseNotes: true
           token: ${{ secrets.MY_GITHUB_TOKEN }}
@@ -131,14 +131,14 @@ jobs:
           token: ${{ secrets.MY_GITHUB_TOKEN }}
           omitBody: true
 
-  release_linux_cu121_glibc231:
+  release_linux_cu124_glibc231:
     runs-on: ubuntu-20.04
     steps:
       - uses: actions/checkout@v4
       - name: setup cuda
         uses: Jimver/[email protected]
         with:
-            cuda: '12.1.0'
+            cuda: '12.4.0'
             method: network
 
       - name: make
@@ -148,29 +148,29 @@ jobs:
         run: |
           sudo apt-get clean
           sudo apt-get install -y cmake
-          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu121.zip
+          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu124.zip
           unzip -qq libtorch.zip
           cmake -DCMAKE_PREFIX_PATH=./libtorch -DCMAKE_BUILD_TYPE=Release .
           make -j
-          mv libgotorch.so libgotorch_gpu.cu121.glibc231.so
+          mv libgotorch.so libgotorch_gpu.cu124.glibc231.so
 
       - name: upload
         uses: ncipollo/release-action@v1
         with:
-          artifacts: ./lib/libgotorch_gpu.cu121.glibc231.so
+          artifacts: ./lib/libgotorch_gpu.cu124.glibc231.so
           allowUpdates: true
           generateReleaseNotes: true
           token: ${{ secrets.MY_GITHUB_TOKEN }}
           omitBody: true
 
-  release_linux_cu124_glibc231:
+  release_linux_cu126_glibc231:
     runs-on: ubuntu-20.04
     steps:
       - uses: actions/checkout@v4
       - name: setup cuda
         uses: Jimver/[email protected]
         with:
-            cuda: '12.4.0'
+            cuda: '12.6.0'
             method: network
 
       - name: make
@@ -180,16 +180,16 @@ jobs:
         run: |
           sudo apt-get clean
           sudo apt-get install -y cmake
-          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu124.zip
+          wget -q -O libtorch.zip https://download.pytorch.org/libtorch/cu126/libtorch-cxx11-abi-shared-with-deps-${{ env.libtorch_version }}%2Bcu126.zip
           unzip -qq libtorch.zip
           cmake -DCMAKE_PREFIX_PATH=./libtorch -DCMAKE_BUILD_TYPE=Release .
           make -j
-          mv libgotorch.so libgotorch_gpu.cu124.glibc231.so
+          mv libgotorch.so libgotorch_gpu.cu126.glibc231.so
 
       - name: upload
         uses: ncipollo/release-action@v1
         with:
-          artifacts: ./lib/libgotorch_gpu.cu124.glibc231.so
+          artifacts: ./lib/libgotorch_gpu.cu126.glibc231.so
           allowUpdates: true
           generateReleaseNotes: true
           token: ${{ secrets.MY_GITHUB_TOKEN }}