Skip to content

Commit

Permalink
Removed mkl from dev deps (#3226)
Browse files Browse the repository at this point in the history
* Removed mkl from dev deps

* Updated tpu gha yaml

* Install mkl for tpu GHA

* fix mkl installation for tpu GHA

* fix tpu gha

* fix tpu gha 2
  • Loading branch information
vfdev-5 authored Apr 1, 2024
1 parent 216b348 commit b828059
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 9 deletions.
22 changes: 14 additions & 8 deletions .github/workflows/tpu-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,16 @@ on:
- "ignite/**"
- "tests/ignite/**"
- "tests/run_tpu_tests.sh"
- "tests/run_code_style.sh"
- "requirements-dev.txt"
- ".github/workflows/tpu-tests.yml"
pull_request:
paths:
- "ignite/**"
- "tests/ignite/**"
- "tests/run_tpu_tests.sh"
- "tests/run_code_style.sh"
- "requirements-dev.txt"
- ".github/workflows/tpu-tests.yml"
workflow_dispatch:

Expand All @@ -32,10 +36,10 @@ jobs:

steps:
- uses: actions/checkout@v4
- name: Set up Python 3.8
- name: Set up Python 3.9
uses: actions/setup-python@v4
with:
python-version: "3.8"
python-version: "3.9"
architecture: "x64"

- name: Get year & week number
Expand All @@ -61,15 +65,17 @@ jobs:
- name: Install Torch XLA and others
run: |
## Install openblas, mkl, gsutil
## Install mkl (alternative approach to https://github.com/pytorch/xla/blob/b0ba29f98a695671972d4a4cc07441014dba2892/.kokoro/common.sh#L31-L32)
sudo apt-get update && sudo apt-get install -y libopenblas-dev libomp5
# mkl version fixed due to https://github.com/pytorch/ignite/issues/2350
pip install mkl==2021.4.0 requests gsutil
pip install mkl==2021.4.0
## Install torch & xla and torchvision
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch-nightly-cp38-cp38-linux_x86_64.whl
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-nightly-cp38-cp38-linux_x86_64.whl
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torchvision-nightly-cp38-cp38-linux_x86_64.whl
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch-nightly-cp39-cp39-linux_x86_64.whl
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-nightly-cp39-cp39-linux_x86_64.whl
pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torchvision-nightly-cp39-cp39-linux_x86_64.whl
# Check installation
python -c "import torch"
## Install test deps and Ignite
pip install -r requirements-dev.txt
Expand Down
1 change: 0 additions & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,5 @@ git+https://github.com/nltk/nltk
# Examples dependencies
pandas
gymnasium
mkl;platform_machine=="x86_64"
# temporary fix: E AttributeError: module 'mpmath' has no attribute 'rational'
mpmath<1.4

0 comments on commit b828059

Please sign in to comment.