From 1e98a061a3be232d81a1e60b4914e2939fe1d593 Mon Sep 17 00:00:00 2001 From: lijian19981118 Date: Thu, 3 Jul 2025 02:03:50 +0800 Subject: [PATCH 1/2] modify CANN, python, pytorch version in requirement table --- .../cv/classfication/Swin-Transformer_tiny/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/README.md b/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/README.md index f138a497ab..ef6409cea7 100644 --- a/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/README.md +++ b/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/README.md @@ -54,9 +54,9 @@ Swin-Transformer是针对于图片处理设计的基于Transformer架构的神 | 配套 | 版本 | 环境准备指导 | | ------------------------------------------------------------ | ------- | ------------------------------------------------------------ | | 固件与驱动 | 1.0.17 | [Pytorch框架推理环境准备](https://www.hiascend.com/document/detail/zh/ModelZoo/pytorchframework/pies) | -| CANN | 6.0.RC1 | - | -| Python | 3.7.5 | - | -| PyTorch | 1.8.0+ | - | +| CANN | 8.1.RC1 | - | +| Python | 3.8 | - | +| PyTorch | 2.1.0 | - | | 说明:Atlas 300I Duo 推理卡请以CANN版本选择实际固件与驱动版本。 | \ | \ | # 快速上手 -- Gitee From e46a013c3038b2ccba933fc2ef1eb88bd15747bb Mon Sep 17 00:00:00 2001 From: lijian19981118 Date: Thu, 3 Jul 2025 06:35:06 +0800 Subject: [PATCH 2/2] update torch version aligned with README --- .../cv/classfication/Swin-Transformer_tiny/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/requirements.txt b/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/requirements.txt index 99d104329b..467abf629d 100644 --- a/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/requirements.txt +++ b/ACL_PyTorch/contrib/cv/classfication/Swin-Transformer_tiny/requirements.txt @@ -8,6 +8,6 @@ Pillow==8.2.0 PyYAML==5.4.1 termcolor==1.1.0 timm==0.3.2 -torch==1.8.0 -torchvision==0.9.0 +torch==2.1.0 +torchvision==0.16.0 yacs==0.1.8 -- Gitee