Skip to content

GitLab

"runtime/rust/examples/vscode:/vscode.git/clone" did not exist on "cdcdce969ec27556d71442a0b4327b6f27811a1d"
Commit 05a24645 authored by chenzk's avatar chenzk
Browse files

Update sf.md

parent cc956a72
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
README.md
View file @ 05a24645
...@@ -21,7 +21,7 @@ MambaVision通过创建没有 SSM 的对称路径来引入一种新颖的混合 ...@@ -21,7 +21,7 @@ MambaVision通过创建没有 SSM 的对称路径来引入一种新颖的混合
## 环境配置 ## 环境配置
### Docker（方法一） ### Docker（方法一）
此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤，以及[光合](https://developer.hpccube.com/tool/)开发者社区深度学习库下载地址 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤，以及[光合](https://developer.sourcefind.cn/tool/)开发者社区深度学习库下载地址
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it --shm-size=1024G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name MambaVision_pytorch <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：a4dd5be0ca23 docker run -it --shm-size=1024G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name MambaVision_pytorch <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：a4dd5be0ca23
...@@ -54,7 +54,7 @@ pip install timm==0.9.0 tensorboardX ...@@ -54,7 +54,7 @@ pip install timm==0.9.0 tensorboardX
### Anaconda（方法三） ### Anaconda（方法三）
此处提供本地配置、编译的详细步骤，例如： 此处提供本地配置、编译的详细步骤，例如：
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.sourcefind.cn/tool/)开发者社区下载安装。
``` ```
#DTK驱动：dtk24.04.1 #DTK驱动：dtk24.04.1
# python：python3.10 # python：python3.10
...@@ -164,7 +164,7 @@ Inference ： ...@@ -164,7 +164,7 @@ Inference ：
[MambaVision-T-1K](https://huggingface.co/nvidia/MambaVision-T-1K) [MambaVision-T-1K](https://huggingface.co/nvidia/MambaVision-T-1K)
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
- https://developer.hpccube.com/codes/modelzoo/mambavision_pytorch - https://developer.sourcefind.cn/codes/modelzoo/mambavision_pytorch
## 参考资料 ## 参考资料
- https://github.com/NVlabs/MambaVision - https://github.com/NVlabs/MambaVision
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