ci: testing with torch=2.0 (#1671)

* testing with `torch=2.0`
* gnu

.azure/gpu-pipeline.yml

@@ -20,19 +20,18 @@ jobs:
  - job: pytest
  strategy:
  matrix:
- 'PyTorch - old':
+ 'PyTorch | old':
  docker-image: 'pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime'
  agent-pool: 'lit-rtx-3090'
  torch-ver: '1.9.0'
- 'PyTorch - LTS':
+ 'PyTorch | 1.X':
  docker-image: 'pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime'
  agent-pool: 'lit-rtx-3090'
  torch-ver: '1.13.1'
- # todo: when docker are live
- #'PyTorch - stable':
- # docker-image: 'pytorch/pytorch:2.0.0-cuda11.6-cudnn8-runtime'
- # agent-pool: 'lit-rtx-3090'
- # torch-ver: '2.0.0'
+ 'PyTorch | 2.X':
+ docker-image: 'pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime'
+ agent-pool: 'lit-rtx-3090'
+ torch-ver: '2.0.0'
  # how long to run the job before automatically cancelling
  timeoutInMinutes: "120"
  # how much time to give 'run always even if cancelled tasks' before stopping them
@@ -47,6 +46,8 @@ jobs:
  TORCH_HOME: "/var/tmp/torch"
  TRANSFORMERS_CACHE: "/var/tmp/huggingface"
  PIP_CACHE_DIR: "/var/tmp/pip"
+ # MKL_THREADING_LAYER: "GNU"
+ MKL_SERVICE_FORCE_INTEL: 1
  # todo: consider unfreeze for master too
  FREEZE_REQUIREMENTS: 1