diff --git a/ACL_PyTorch/built-in/cv/SAM/README.md b/ACL_PyTorch/built-in/cv/SAM/README.md index 7fee7963cac6af830aa8d1b81c4e70c0be807e27..4ca3afa1f55cc9977408741eddf3a8e66c4a0e2a 100644 --- a/ACL_PyTorch/built-in/cv/SAM/README.md +++ b/ACL_PyTorch/built-in/cv/SAM/README.md @@ -61,11 +61,11 @@ SAM 首先会自动分割图像中的所有内容,但是如果你需要分割 | 配套 | 版本 | 环境准备指导 | | ---- | ---- | ---- | -| 固件与驱动 | 24.0.T1.B010 | [Pytorch框架推理环境准备](https://www.hiascend.com/document/detail/zh/ModelZoo/pytorchframework/pies) | -| CANN | 8.0.RC1.B060 | - | -| MindIE | 1.0.RC1.B060 | - | -| Python | 3.10.13(MindIE 要求 Python 3.10) | - | -| PyTorch | 1.13.1 | - | +| 固件与驱动 | 25.2.RC1 | [Pytorch框架推理环境准备](https://www.hiascend.com/document/detail/zh/ModelZoo/pytorchframework/pies) | +| CANN | 8.2.RC1 | - | +| MindIE | 2.1.RC1 | - | +| Python | 3.11.10 | - | +| PyTorch | 2.1.0 | - | | 说明:Atlas 300I Duo 推理卡请以CANN版本选择实际固件与驱动版本。 | \ | \ | ## 3. 快速上手 @@ -78,7 +78,7 @@ cd ModelZoo-PyTorch/ACL_PyTorch/built-in/cv/SAM git clone https://github.com/facebookresearch/segment-anything.git cd segment-anything git reset --hard 6fdee8f2727f4506cfbbe553e23b895e27956588 -patch -p2 < segment_anything_diff.patch +patch -p2 < ../segment_anything_diff.patch pip3 install -e . cd .. ``` diff --git a/ACL_PyTorch/built-in/cv/SAM/requirements.txt b/ACL_PyTorch/built-in/cv/SAM/requirements.txt index 22d75b25e81284364400b6ce7b3db738694cefd1..6722adb464f8d1cbc54c229ff0e6e974124339b4 100644 --- a/ACL_PyTorch/built-in/cv/SAM/requirements.txt +++ b/ACL_PyTorch/built-in/cv/SAM/requirements.txt @@ -1,14 +1,15 @@ -torch==1.13.1 -torchvision==0.14.1 -torchaudio==0.13.1 +torch==2.1.0 +torch_npu==2.1.0.post17.dev20250905 +torchvision==0.16.0 +torchaudio==2.1.0 decorator scipy attrs psutil -numpy -opencv-python +numpy==1.26.0 +opencv-python==4.11.0.86 pycocotools matplotlib +onnx==1.16.1 onnxruntime -onnx onnxsim==0.4.17 \ No newline at end of file