diff --git "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" index 0a2b7cd1b3d7ab6977cd8e74fbc5bd209ad984f7..57da05ef9f375d5c0ed14af9fd10008787c852dc 100644 --- "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" +++ "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" @@ -35,6 +35,14 @@ - 需完成CANN开发或运行环境的安装,具体操作请参考《CANN 软件安装指南》。 + | AscendPytorch版本 | CANN版本 | 支持PyTorch版本 | + | ----------------- | ---------- | -------------------------------------- | + | 2.0.2 | CANN 5.0.2 | 1.5.0 | + | 2.0.3 | CANN 5.0.3 | 1.5.0/1.8.1(1.8.1仅支持resnet50模型) | + | 2.0.4 | CANN 5.0.4 | 1.5.0/1.8.1(1.8.1仅支持resnet50模型) | + + + - 需安装3.12.0以上版本的CMake,安装方法请参考[CMake安装方法](#CMake安装方法md)。 - 需确保已安装7.3.0以上版本的gcc,7.3.0版本gcc具体安装及使用方式请参考[安装7.3.0版本gcc](#安装7-3-0版本gccmd)。 @@ -74,7 +82,7 @@ pip3 install wheel ``` -3. 获取PyTorch源代码。 +3. 获取PyTorch源代码。 1. 运行如下命令,获取适配昇腾AI处理器的PyTorch源代码,并切换到所需的分支。