.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -76,29 +76,24 @@ jobs:
         with:
           name: pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64
           path: pytorch/torchcodec/dist/
-      - name: Setup conda env
-        uses: conda-incubator/setup-miniconda@v2
+      - name: Setup miniconda using test-infra
+        uses: pytorch/test-infra/.github/actions/setup-miniconda@main
         with:
-          auto-update-conda: true
-          miniconda-version: "latest"
-          activate-environment: test
-          python-version: ${{ matrix.python-version }}
+          python-version: "3.9"
       - name: Check env
         run: |
-          env
-          conda info
+          ${CONDA_RUN} env
+          ${CONDA_RUN} conda info
       - name: Update pip
-        run: python -m pip install --upgrade pip
+        run: ${CONDA_RUN} python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          conda activate test
-          python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
+          ${CONDA_RUN} python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
       - name: Install torchcodec from the wheel
         run: |
           wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`
           echo Installing $wheel_path
-          conda activate test
-          python -m pip install $wheel_path -vvv
+          ${CONDA_RUN} python -m pip install $wheel_path -vvv
 
       - name: Check out repo
         uses: actions/checkout@v3