Skip to content
This repository was archived by the owner on Aug 28, 2025. It is now read-only.

Commit 1143a73

Browse files
authored
update the GPU env to PT 1.10 (#145)
* update to PT 1.10 * builds * metrics * pygame
1 parent d52eada commit 1143a73

File tree

12 files changed

+14
-12
lines changed

12 files changed

+14
-12
lines changed

.actions/assistant.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def _parse_requirements(folder: str) -> Tuple[str, str]:
247247
folder: path to the folder with python script, meta and artefacts
248248
"""
249249
meta = AssistantCLI._load_meta(folder)
250-
req = meta.get("requirements", [])
250+
reqs = meta.get("requirements", [])
251251

252252
meta_pip_args = {
253253
k.replace(AssistantCLI._META_PIP_KEY, ""): v
@@ -262,7 +262,7 @@ def _parse_requirements(folder: str) -> Tuple[str, str]:
262262
arg = arg % _RUNTIME_VERSIONS
263263
pip_args.append(f"--{pip_key} {arg}")
264264

265-
return " ".join(req), " ".join(pip_args)
265+
return " ".join([f'"{req}"' for req in reqs]), " ".join(pip_args)
266266

267267
@staticmethod
268268
def _bash_download_data(folder: str) -> List[str]:

.azure/ipynb-publish.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ jobs:
2121
# this need to have installed docker in the base image...
2222
container:
2323
# base ML image: mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04
24-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.8"
24+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.10"
2525
# image: "pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime"
2626
options: "-it --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
2727

.azure/ipynb-tests.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
# this need to have installed docker in the base image...
1818
container:
1919
# base ML image: mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04
20-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.8"
20+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.10"
2121
# image: "pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime"
2222
options: "-it --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
2323

lightning_examples/augmentation_kornia/.meta.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: PL/Kornia team
33
created: 2021-06-11
44
updated: 2021-06-16
55
license: CC BY-SA
6-
build: 3
6+
build: 4
77
tags:
88
- Image
99
description: |

lightning_examples/barlow-twins/.meta.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ author: Ananya Harsh Jha (ananya@pytorchlightning.ai)
33
created: 2021-09-19
44
updated: 2021-09-20
55
license: CC BY-SA
6+
build: 1
67
tags:
78
- Image
89
- Self-Supervised

lightning_examples/barlow-twins/barlow_twins.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
import torchvision.transforms.functional as VisionF
1818
from pytorch_lightning import Callback, LightningModule, Trainer
1919
from pytorch_lightning.callbacks import ModelCheckpoint
20-
from pytorch_lightning.metrics.functional import accuracy
2120
from torch.utils.data import DataLoader
21+
from torchmetrics.functional import accuracy
2222
from torchvision.datasets import CIFAR10
2323
from torchvision.models.resnet import resnet18
2424
from torchvision.utils import make_grid

lightning_examples/basic-gan/.meta.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: PL team
33
created: 2020-12-21
44
updated: 2021-06-16
55
license: CC BY-SA
6-
build: 4
6+
build: 5
77
tags:
88
- Image
99
description: |

lightning_examples/cifar10-baseline/.meta.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: PL team
33
created: 2020-12-21
44
updated: 2021-06-16
55
license: CC BY-SA
6-
build: 2
6+
build: 3
77
tags:
88
- Image
99
description: >

lightning_examples/datamodules/.meta.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: PL team
33
created: 2020-12-21
44
updated: 2021-06-07
55
license: CC BY-SA
6-
build: 2
6+
build: 3
77
description: This notebook will walk you through how to start using Datamodules. With
88
the release of `pytorch-lightning` version 0.9.0, we have included a new class called
99
`LightningDataModule` to help you decouple data related hooks from your `LightningModule`.

lightning_examples/mnist-hello-world/.meta.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: PL team
33
created: 2020-12-21
44
updated: 2021-06-16
55
license: CC BY-SA
6-
build: 2
6+
build: 3
77
tags:
88
- Image
99
description: In this notebook, we'll go over the basics of lightning by preparing

0 commit comments

Comments
 (0)