# chat_demo 技服智能问答服务 ## 环境配置 ### Docker(方式一) -v 路径、docker_name和imageID根据实际情况修改 ```bash docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310 docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=80G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash # 加载运行环境变量 unzip dtk-cuda.zip mv ./cuda /opt/dtk/ source /opt/dtk/cuda/env.sh # 下载fastllm库 git clone http://developer.hpccube.com/codes/OpenDAS/fastllm.git # 编译fastllm cd fastllm mkdir build cd build cmake .. make -j # 编译完成后,可以使用如下命令安装简易python工具包 cd tools # 这时在fastllm/build/tools目录下 python setup.py install cd /path/of/chat_demo pip install faiss-1.7.2_dtk24.04_gitb7348e7df780-py3-none-any.whl pip install -r requirements.txt ``` ### Dockerfile(方式二) ``` docker build -t chat_demo:latest . docker run -dit --network=host --name=chat_demo --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 chat_demo:latest docker exec -it chat_demo /bin/bash # 其他步骤同上面的Docker(方式一) ``` ### Conda(方法三) 关于本项目DCU显卡所需的工具包、深度学习库等均可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 ```bash DTK驱动: dtk24.04 python: python3.10 torch: 2.1.0 ``` `Tips:以上dtk驱动、python、deepspeed等工具版本需要严格一一对应。` 3. 其它依赖库参照requirements.txt安装: ``` pip install faiss-1.7.2_dtk24.04_gitb7348e7df780-py3-none-any.whl pip install -r requirements.txt ```