Skip to content

Commit 74bdf24

Browse files
BordaBorda
committed
image
1 parent 55ce170 commit 74bdf24

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

‎.lightning/workflows/benchmark.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ timeout: "90" # minutes
88
parametrize:
99
matrix:
1010
PACKAGE_NAME: ["fabric", "pytorch"]
11+
image: "nvidia/cuda:12.6.3-devel-ubuntu22.04"
1112
machine: "L4_X_2"
1213
env:
1314
python_version: "3.12"

0 commit comments

Comments
 (0)