Skip to content

Commit 41089fd

Browse files
committed
fix: 使用CPU版PyTorch减小包体积避免超过2GB限制
1 parent 9ec56d5 commit 41089fd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

.github/workflows/build-executables.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ jobs:
2626
run: |
2727
python -m pip install --upgrade "pip<24.1"
2828
pip install pyinstaller
29-
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu121
29+
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
3030
pip install omegaconf==2.0.6 --no-deps
3131
pip install PyYAML antlr4-python3-runtime hydra-core
3232
pip install fairseq==0.12.2 --no-deps
@@ -115,7 +115,7 @@ jobs:
115115
run: |
116116
python -m pip install --upgrade "pip<24.1"
117117
pip install pyinstaller
118-
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu121
118+
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
119119
pip install omegaconf==2.0.6 --no-deps
120120
pip install PyYAML antlr4-python3-runtime hydra-core
121121
pip install fairseq==0.12.2 --no-deps

0 commit comments

Comments
 (0)