Install PyTorch from PyPi (#437)

Author: mthrok

.github/workflows/_build_linux.yml

@@ -89,12 +89,20 @@ jobs:
 
       - name: Unit test
         run: |
+          # Install SPDL
           pip install $(find "${HOME}/package" -name '*.whl' -depth -maxdepth 1)
+
+          # Install PyTorch and others
+          pip install numpy torch --index-url https://download.pytorch.org/whl/cpu
+          conda install -q numba pytest
+
+          # Install FFmpeg
           # Note: Somehow FFmepg 5.1 does not install libiconv so specifying it here
           conda install -q -c conda-forge "ffmpeg==${{ matrix.ffmpeg-version }}" libiconv
-          conda install -q -c pytorch numpy pytest pytorch numba
           # For some reason, the dynamic libraries are not found.
           export "LD_LIBRARY_PATH=${CONDA_PREFIX}/lib"
+
+          # Run test
           pytest -v \
               tests/spdl_unittest/io/ \
               tests/spdl_unittest/dataloader/

.github/workflows/_build_linux_cuda.yml

@@ -114,13 +114,20 @@ jobs:
 
       - name: Unit test
         run: |
+          # Install SPDL
           pip install $(find "${HOME}/package" -name '*.whl' -depth -maxdepth 1)
+
+          # Install PyTorch and others
+          pip install pytest
+          pip install numpy nvidia-cuda-runtime-cu12 torch --index-url https://download.pytorch.org/whl/cu126
+
+          # Install FFmpeg
           conda install -q -c conda-forge "ffmpeg==${{ matrix.ffmpeg-version }}"
           # For some reason, the FFmpeg dynamic libraries are not found.
           export "LD_LIBRARY_PATH=${CONDA_PREFIX}/lib"
           # TODO: clean up the logic for versions
-          pip install nvidia-cuda-runtime-cu12 torch --index-url https://download.pytorch.org/whl/cu126
-          pip install pytest
+
+          # Run test
           pytest -v \
               tests/spdl_unittest/cuda/ \
               tests/spdl_unittest/io/

.github/workflows/_build_macos.yml

@@ -78,9 +78,15 @@ jobs:
 
       - name: Unit test
         run: |
+          # Install SPDL
           pip install $(find "${HOME}/package" -name '*.whl' -depth -maxdepth 1)
+
+          # Install PyTorch and others
+          pip install torch numpy numba pytest
+
+          # Install FFmpeg
           conda install -q -c conda-forge "ffmpeg==${{ matrix.ffmpeg-version }}"
-          conda install -q -c pytorch numpy pytest pytorch numba
+
           pytest -v \
               tests/spdl_unittest/io/ \
               tests/spdl_unittest/dataloader/