初始化项目,由ModelHub XC社区提供模型
Model: changkaiyan/ChipGPT-Llama2-SFT-7B Source: Original Platform
This commit is contained in:
18
alpaca-lora/Dockerfile
Normal file
18
alpaca-lora/Dockerfile
Normal file
@@ -0,0 +1,18 @@
|
||||
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
||||
|
||||
ARG DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
RUN apt-get update && apt-get install -y \
|
||||
git \
|
||||
curl \
|
||||
software-properties-common \
|
||||
&& add-apt-repository ppa:deadsnakes/ppa \
|
||||
&& apt install -y python3.10 \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
WORKDIR /workspace
|
||||
COPY requirements.txt requirements.txt
|
||||
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 \
|
||||
&& python3.10 -m pip install -r requirements.txt \
|
||||
&& python3.10 -m pip install numpy --pre torch --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cu118
|
||||
COPY . .
|
||||
ENTRYPOINT [ "python3.10"]
|
||||
Reference in New Issue
Block a user