Commit 1aa9c13f authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix GPU tests with speech

parent 2295d783
...@@ -51,8 +51,9 @@ jobs: ...@@ -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: ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment