From 79a22eaee0813d5a0e4dc82ebd4a43071399c899 Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Tue, 1 Nov 2022 16:26:22 -0700 Subject: [PATCH] Upperbound PTL (#5302) * Upperbound PTL Signed-off-by: smajumdar * Upperbound PTL Signed-off-by: smajumdar * Upperbound PTL Signed-off-by: smajumdar * Upperbound PTL Signed-off-by: smajumdar Signed-off-by: smajumdar --- ci.groovy | 4 +++- requirements/requirements_lightning.txt | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/ci.groovy b/ci.groovy index fb4b790b7fb9..8811fda98fbe 100644 --- a/ci.groovy +++ b/ci.groovy @@ -27,11 +27,12 @@ spec: restartPolicy: Never backoffLimit: 4 tty: true - shm-size: 8g + shm-size: 32g nodeSelector: kubernetes.io/os: linux nvidia.com/gpu_type: "Tesla_T4x4" nvidia.com/node_type: gpu_tester + nvidia.com/driver_version: "510.20" """ ) { node(POD_LABEL) { @@ -59,6 +60,7 @@ spec: stage('Code Style') { sh "apt-get update && \ apt-get install -y bc && \ + nvidia-smi && \ pip install -r requirements/requirements_test.txt && \ python setup.py style && ls -l /testdata/TestData && ln -s /testdata/TestData /home/TestData && \ ls -l /home && ls -l /home/TestData" diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt index ea1ab05b78fa..b883f847b51f 100644 --- a/requirements/requirements_lightning.txt +++ b/requirements/requirements_lightning.txt @@ -1,4 +1,4 @@ -pytorch-lightning>=1.7.0 +pytorch-lightning>=1.7.0,<1.8 torchmetrics>=0.4.1rc0 transformers>=4.0.1,<=4.21.2 webdataset>=0.1.48,<=0.1.62