Commit 2a4034b4 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 7b4fb13b
......@@ -36,11 +36,12 @@ mv inpaint-anything_pytorch inpaint-anything # 去框架名后缀
<!-- 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 -->
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10 # 本镜像imageID为:2f1f619d0182
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=16G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=16G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --network=host --name docker_name imageID bash
cd /your_code_path/inpaint-anything
pip install -e segment_anything
pip install diffusers transformers accelerate scipy safetensors
pip install -r requirements.txt
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
# 如果pip安装下载慢建议多尝试更换镜像源(下同)
```
### Dockerfile(方法二)
<!-- 此处提供dockerfile的使用方法 -->
......@@ -68,7 +69,7 @@ pytorch: 2.1.0
```
pip install -e segment_anything
pip install diffusers transformers accelerate scipy safetensors
pip install -r requirement.txt
pip install -r requirement.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
```
## 数据集
......
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