Update wheels-generator.yml

OutofAi · May 11, 2024 · ce92b40 · ce92b40
1 parent 5054026
commit ce92b40
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/.github/workflows/wheels-generator.yml b/.github/workflows/wheels-generator.yml
@@ -208,6 +208,7 @@ jobs:
           source $HOME/miniconda/etc/profile.d/conda.sh
           conda create -n myenv python=3.10 -y
           conda activate myenv
+          conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia
         shell: bash
 
       - uses: actions/checkout@v3
@@ -227,14 +228,6 @@ jobs:
           pip install setuptools wheel
         shell: bash
 
-      - name: Install Pytorch
-        run: |
-          source $HOME/miniconda/etc/profile.d/conda.sh
-          conda activate myenv
-          python --version
-          pip install torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
-        shell: bash
-
       - name: Generate Wheels
         run: |
           source $HOME/miniconda/etc/profile.d/conda.sh