autogluon icon indicating copy to clipboard operation
autogluon copied to clipboard

Published 20 hours ago • verified publisher icon autogluon

Upgrade to torch 2.2 and update CI images

Open prateekdesai04 opened this issue 10 months ago • 2 comments

Issue #, if available:

Description of changes: This is a PR similar to https://github.com/autogluon/autogluon/pull/4118 But includes the updated image changes, and these images are also built and pushed to AG's ECR repo which will be used by the CI. Pre-this PR CI was having Torch 2.1 + cu121 for GPU and Torch 2.1 for CPU Post-this PR CI has Torch 2.2 + cu12.1 for GPU and Torch 2.2 for CPU

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

prateekdesai04 avatar Apr 22 '24 20:04 prateekdesai04

Previous CI Run Current CI Run
botocore==1.34.88 botocore==1.34.89
boto3==1.34.88 boto3==1.34.89
botocore==1.34.88 botocore==1.34.89
boto3==1.34.88 boto3==1.34.89

yinweisu avatar Apr 22 '24 20:04 yinweisu

Previous CI Run Current CI Run
botocore==1.34.107 botocore==1.34.108
boto3==1.34.107 boto3==1.34.108
botocore==1.34.107 botocore==1.34.108
boto3==1.34.107 boto3==1.34.108

yinweisu avatar May 17 '24 20:05 yinweisu

Job PR-4123-52d445f is done. Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4123/52d445f/index.html

github-actions[bot] avatar May 21 '24 00:05 github-actions[bot]

Previous CI Run Current CI Run
pytorch-lightning==2.2.4 pytorch-lightning==2.1.4
torchvision==0.17.2 torchvision==0.16.2
annotated-types==0.6.0 annotated-types==0.7.0
triton==2.2.0 triton==2.1.0
torch==2.2.2 torch==2.1.2
nvidia-nccl-cu12==2.19.3 nvidia-nccl-cu12==2.18.1
transformers==4.39.3 transformers==4.38.2
pytorch-lightning==2.2.4 pytorch-lightning==2.1.4
torchvision==0.17.2 torchvision==0.16.2
annotated-types==0.6.0 annotated-types==0.7.0
triton==2.2.0 triton==2.1.0
torch==2.2.2 torch==2.1.2
nvidia-nccl-cu12==2.19.3 nvidia-nccl-cu12==2.18.1
transformers==4.39.3 transformers==4.38.2

yinweisu avatar May 21 '24 00:05 yinweisu

Previous CI Run Current CI Run

yinweisu avatar May 21 '24 17:05 yinweisu

Previous CI Run Current CI Run
botocore==1.34.109 botocore==1.34.110
boto3==1.34.109 boto3==1.34.110
botocore==1.34.109 botocore==1.34.110
boto3==1.34.109 boto3==1.34.110

yinweisu avatar May 21 '24 20:05 yinweisu

Job PR-4123-54e3e21 is done. Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4123/54e3e21/index.html

github-actions[bot] avatar May 22 '24 00:05 github-actions[bot]