Upperbound PTL (NVIDIA#5302)

* Upperbound PTL Signed-off-by: smajumdar <[email protected]> * Upperbound PTL Signed-off-by: smajumdar <[email protected]> * Upperbound PTL Signed-off-by: smajumdar <[email protected]> * Upperbound PTL Signed-off-by: smajumdar <[email protected]> Signed-off-by: smajumdar <[email protected]> Signed-off-by: Hainan Xu <[email protected]>
hainan-xv · Nov 29, 2022 · ed76d9a · ed76d9a
1 parent bbf1c02
commit ed76d9a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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