diff --git a/.github/workflows/ci_test-conda.yml b/.github/workflows/ci_test-conda.yml index e464e6778f..fd841b38cb 100644 --- a/.github/workflows/ci_test-conda.yml +++ b/.github/workflows/ci_test-conda.yml @@ -50,7 +50,6 @@ jobs: run: | # NOTE: run coverage on tests does not propagare faler status for Win, https://github.com/nedbat/coveragepy/issues/1003 python -m pytest pytorch_lightning tests -v --durations=50 --junitxml=junit/test-results-${{ runner.os }}-${{ matrix.python-version }}-${{ matrix.requires }}.xml - # coverage report shell: bash -l {0} - name: Upload pytest test results diff --git a/dockers/base-conda/Dockerfile b/dockers/base-conda/Dockerfile index e310bb31e4..37a9d3d339 100644 --- a/dockers/base-conda/Dockerfile +++ b/dockers/base-conda/Dockerfile @@ -61,6 +61,7 @@ ENV \ PATH="/root/miniconda3/bin:$PATH" \ LD_LIBRARY_PATH="/root/miniconda3/lib:$LD_LIBRARY_PATH" \ CUDA_TOOLKIT_ROOT_DIR="/usr/local/cuda" \ + MKL_THREADING_LAYER=GNU \ HOROVOD_GPU_OPERATIONS=NCCL \ HOROVOD_WITH_PYTORCH=1 \ HOROVOD_WITHOUT_TENSORFLOW=1 \