From 78d3d842f6efee2e78b1d86da197d8684f2f1278 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E4=B9=BE=E4=B9=BE?= Date: Wed, 27 Aug 2025 11:20:10 +0800 Subject: [PATCH] support_torch26 --- model_examples/OpenDWM/README.md | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/model_examples/OpenDWM/README.md b/model_examples/OpenDWM/README.md index 69294c6b..ce49c54a 100644 --- a/model_examples/OpenDWM/README.md +++ b/model_examples/OpenDWM/README.md @@ -65,6 +65,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 | 三方库 | 支持版本 | | :-----: | :------: | | PyTorch | 2.5.1 | +| PyTorch | 2.6.0 | ### 安装昇腾环境 @@ -76,7 +77,7 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 | :---------------: | :------: | | FrameworkPTAdapter | 7.1.0 | | CANN | 8.2.RC1 | -| Python | >=3.9 | +| Python | 3.9 | 1. 激活 CANN 环境 将 CANN 包目录记作 cann_root_dir，执行以下命令以激活环境 @@ -108,11 +109,23 @@ OpenDWM是一种统一的多视角驾驶视频生成框架。通过融合单/多 将模型根目录记作 `model-root-path` + + 5. 安装模型相关的依赖项。 - + 根据已安装的 PyTorch 版本，选择对应版本的 torchvision 进行安装： + * 若 PyTorch 版本为 2.5.1： ``` python -m pip install torchvision==0.20.1 + ``` + * 若 PyTorch 版本为 2.6.0： + ``` + python -m pip install torchvision==0.21.0 + ``` + + 请确保 torch 与 torchvision 版本兼容。可通过 `python -c "import torch; print(torch.__version__)"` 查看当前 PyTorch 版本。 + + ``` # 安装其他依赖项 python -m pip install -r requirements.txt -- Gitee