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 # 去框架名后缀 ...@@ -36,11 +36,12 @@ mv inpaint-anything_pytorch inpaint-anything # 去框架名后缀
<!-- 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 --> <!-- 此处提供[光源](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 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 cd /your_code_path/inpaint-anything
pip install -e segment_anything pip install -e segment_anything
pip install diffusers transformers accelerate scipy safetensors 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(方法二)
<!-- 此处提供dockerfile的使用方法 --> <!-- 此处提供dockerfile的使用方法 -->
...@@ -68,7 +69,7 @@ pytorch: 2.1.0 ...@@ -68,7 +69,7 @@ pytorch: 2.1.0
``` ```
pip install -e segment_anything pip install -e segment_anything
pip install diffusers transformers accelerate scipy safetensors 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