diff --git a/.github/workflows/amd.yml b/.github/workflows/amd.yml index 91e8825b83e1..6847180e97b0 100644 --- a/.github/workflows/amd.yml +++ b/.github/workflows/amd.yml @@ -38,6 +38,10 @@ jobs: sudo apt-get update sudo apt-get install -y libaio-dev + - name: Python environment + run: | + pip list + - name: Install transformers run: | git clone https://github.com/huggingface/transformers diff --git a/.github/workflows/nv-lightning-v100.yml b/.github/workflows/nv-lightning-v100.yml index 8fb38525a8c7..bfdf9bc06eab 100644 --- a/.github/workflows/nv-lightning-v100.yml +++ b/.github/workflows/nv-lightning-v100.yml @@ -34,11 +34,17 @@ jobs: pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" + + - name: Python environment + run: | + pip list + - name: Install deepspeed run: | pip uninstall --yes deepspeed pip install .[dev,autotuning] ds_report + - name: PyTorch Lightning Tests run: | if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index e38a7094ebae..d9c14ca440c4 100644 --- a/.github/workflows/nv-torch-latest-v100.yml +++ b/.github/workflows/nv-torch-latest-v100.yml @@ -45,6 +45,10 @@ jobs: pip uninstall --yes transformers pip install . + - name: Python environment + run: | + pip list + - name: Install deepspeed run: | pip uninstall --yes deepspeed diff --git a/.github/workflows/nv-torch-nightly-v100.yml b/.github/workflows/nv-torch-nightly-v100.yml index 26941bb29c92..bb53bef5e240 100644 --- a/.github/workflows/nv-torch-nightly-v100.yml +++ b/.github/workflows/nv-torch-nightly-v100.yml @@ -38,6 +38,10 @@ jobs: pip uninstall --yes transformers pip install . + - name: Python environment + run: | + pip list + - name: Install deepspeed run: | pip uninstall --yes deepspeed diff --git a/.github/workflows/nv-torch12-p40.yml b/.github/workflows/nv-torch12-p40.yml index ff6d4b481746..e61979047d93 100644 --- a/.github/workflows/nv-torch12-p40.yml +++ b/.github/workflows/nv-torch12-p40.yml @@ -35,6 +35,10 @@ jobs: python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" + - name: Python environment + run: | + pip list + - name: Install transformers run: | git clone https://github.com/huggingface/transformers diff --git a/.github/workflows/nv-torch18-v100.yml b/.github/workflows/nv-torch18-v100.yml index abea2e06ea41..abe941e527b7 100644 --- a/.github/workflows/nv-torch18-v100.yml +++ b/.github/workflows/nv-torch18-v100.yml @@ -45,6 +45,10 @@ jobs: pip uninstall --yes transformers pip install . + - name: Python environment + run: | + pip list + - name: Install deepspeed run: | pip uninstall --yes deepspeed diff --git a/.github/workflows/nv-transformers-v100.yml b/.github/workflows/nv-transformers-v100.yml index 9c0ce2757292..97e8eea35aa0 100644 --- a/.github/workflows/nv-transformers-v100.yml +++ b/.github/workflows/nv-transformers-v100.yml @@ -34,11 +34,17 @@ jobs: pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" + + - name: Python environment + run: | + pip list + - name: Install deepspeed run: | pip uninstall --yes deepspeed pip install .[dev,autotuning] ds_report + - name: HF transformers tests run: | if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi @@ -50,5 +56,8 @@ jobs: # scipy/sklearn required for tests, using the 'dev' extra forces torch re-install pip install .[testing] # find reqs used in ds integration tests - find examples/pytorch -regextype posix-egrep -regex '.*(language-modeling|question-answering|summarization|image-classification|text-classification|translation).*/requirements.txt' -exec pip install -r {} \; + find examples/pytorch -regextype posix-egrep -regex '.*(language-modeling|question-answering|summarization|image-classification|text-classification|translation).*/requirements.txt' -exec grep -v 'torch' {} \; | xargs -I {} pip install --upgrade {} + # force protobuf version due to issues + pip install "protobuf<4.21.0" + pip list TORCH_EXTENSIONS_DIR=./torch-extensions RUN_SLOW=1 pytest --color=yes --durations=0 --verbose tests/deepspeed