Skip to content

Commit 8f74454

Browse files
authored
Fix CI pytorch install (#284)
1 parent 5535106 commit 8f74454

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

.github/workflows/unix-openmpi.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ jobs:
2828
- shell: bash -l {0}
2929
name: Install dependencies
3030
run: |
31-
conda install numpy pandas pytorch cpuonly -c pytorch
31+
conda install -c conda-forge pytorch-cpu
32+
conda install -c pytorch numpy pandas
3233
conda install -c conda-forge mpi4py openmpi=5.*
3334
pip install .[test]
3435
- shell: bash -l {0}

.github/workflows/unix.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ jobs:
2828
- shell: bash -l {0}
2929
name: Install dependencies
3030
run: |
31-
conda install numpy pandas pytorch cpuonly -c pytorch
31+
conda install -c conda-forge pytorch-cpu
32+
conda install -c pytorch numpy pandas
3233
conda install -c conda-forge mpi4py mpich
3334
pip install .[test]
3435
- shell: bash -l {0}

0 commit comments

Comments
 (0)