SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix GPU tests with speech

Browse Source
This commit is contained in:
Sylvain Gugger
2021-03-10 12:51:06 -05:00
parent 2295d783d5
commit 1aa9c13f70
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/self-push.yml vendored
View File

@@ -51,8 +51,9 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
source .env/bin/activate source .env/bin/activate
sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece,speech]
pip install git+https://github.com/huggingface/datasets pip install git+https://github.com/huggingface/datasets
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
@@ -197,8 +198,9 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
source .env/bin/activate source .env/bin/activate
sudo apt-get -y update && sudo apt-get install -y libsndfile1-dev
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece,speech]
pip install git+https://github.com/huggingface/datasets pip install git+https://github.com/huggingface/datasets
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks