Skip to content

GitLab

"...composable_kernel_rocm.git" did not exist on "bd811e2c2055d8da8d13d7987c40f133943de6a0"
Commit 5082bb1e authored by chenzk's avatar chenzk
Browse files

v1.2.3

parent bed86591
Pipeline #583 failed with stage
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
README.md
View file @ 5082bb1e
...@@ -12,6 +12,9 @@ Vision Transformer先将图像用卷积进行分块以降低计算量，再对 ...@@ -12,6 +12,9 @@ Vision Transformer先将图像用卷积进行分块以降低计算量，再对
![img](./doc/attention.png) ![img](./doc/attention.png)
## 环境配置 ## 环境配置
```
mv megatron-deepspeed-vit_pytorch megatron-deepspeed-vit # 去框架名后缀
```
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py38-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py38-latest
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment