run K100_AI

README.md

@@ -24,8 +24,15 @@ VisualGLM-6B 由 SwissArmyTransformer(简称sat) 库训练，这是一个支持T
 ### Docker（方法一）
 ```
 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
-docker run -it -v /opt/hyhal:/opt/hyhal:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
+docker run -it -v /opt/hyhal:/opt/hyhal:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash\
 ```
+
+安装环境
+```
+docker exec -it contailer_ID bash
+pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
+```
+
 ### Dockerfile（方法二）
 此处提供dockerfile的使用方法， dockerfile在docker文件夹中
 ```