-
Notifications
You must be signed in to change notification settings - Fork 7.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
1404811
commit 68bdec1
Showing
2 changed files
with
78 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
# https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages | ||
name: Create and publish a Docker image for ChatGLM support | ||
|
||
on: | ||
push: | ||
branches: | ||
- 'master' | ||
|
||
env: | ||
REGISTRY: ghcr.io | ||
IMAGE_NAME: ${{ github.repository }}_jittorllms | ||
|
||
jobs: | ||
build-and-push-image: | ||
runs-on: ubuntu-latest | ||
permissions: | ||
contents: read | ||
packages: write | ||
|
||
steps: | ||
- name: Checkout repository | ||
uses: actions/checkout@v3 | ||
|
||
- name: Log in to the Container registry | ||
uses: docker/login-action@v2 | ||
with: | ||
registry: ${{ env.REGISTRY }} | ||
username: ${{ github.actor }} | ||
password: ${{ secrets.GITHUB_TOKEN }} | ||
|
||
- name: Extract metadata (tags, labels) for Docker | ||
id: meta | ||
uses: docker/metadata-action@v4 | ||
with: | ||
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }} | ||
|
||
- name: Build and push Docker image | ||
uses: docker/build-push-action@v4 | ||
with: | ||
context: . | ||
push: true | ||
file: docs/GithubAction+JittorLLMs | ||
tags: ${{ steps.meta.outputs.tags }} | ||
labels: ${{ steps.meta.outputs.labels }} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,34 @@ | ||
# 从NVIDIA源,从而支持显卡运损(检查宿主的nvidia-smi中的cuda版本必须>=11.3) | ||
FROM nvidia/cuda:11.3.1-runtime-ubuntu20.04 | ||
ARG useProxyNetwork='' | ||
RUN apt-get update | ||
RUN apt-get install -y curl proxychains curl g++ | ||
RUN apt-get install -y git python python3 python-dev python3-dev --fix-missing | ||
|
||
# use python3 as the system default python | ||
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.8 | ||
|
||
# 下载pytorch | ||
RUN python3 -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu113 | ||
|
||
# 下载分支 | ||
WORKDIR /gpt | ||
RUN git clone https://github.com/binary-husky/chatgpt_academic.git -b jittor | ||
WORKDIR /gpt/chatgpt_academic | ||
RUN python3 -m pip install -r requirements.txt | ||
RUN python3 -m pip install -r request_llm/requirements_chatglm.txt | ||
RUN python3 -m pip install -r request_llm/requirements_newbing.txt | ||
RUN python3 -m pip install -r request_llm/requirements_jittorllms.txt -i https://pypi.jittor.org/simple -I | ||
|
||
# 下载JittorLLMs | ||
RUN git clone https://github.com/binary-husky/JittorLLMs.git --depth 1 request_llm/jittorllms | ||
|
||
# 禁用缓存,确保更新代码 | ||
ADD "https://www.random.org/cgi-bin/randbyte?nbytes=10&format=h" skipcache | ||
RUN git pull | ||
|
||
# 预热Tiktoken模块 | ||
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()' | ||
|
||
# 启动 | ||
CMD ["python3", "-u", "main.py"] |