-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDockerfile
74 lines (54 loc) · 1.65 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#######################################
# Docker Environment to run local LLM #
#######################################
FROM ubuntu:20.04
###############
# Environment #
###############
ENV LANG=en_US.utf-8
ENV LC_ALL=en_US.utf-8
ENV NODE_VERSION=18.15.0
ENV NVM_DIR=/root/.nvm
ENV PATH="/root/.nvm/versions/node/v${NODE_VERSION}/bin/:${PATH}"
ENV DEBIAN_FRONTEND="noninteractive"
###########
# General #
###########
RUN apt-get update && \
apt-get install -y curl wget git python3-pip python3.10.9
# python3.8-distutils python3.8-venv virtualenv
RUN ln -s /usr/bin/python3 /usr/bin/python & \
ln -s /usr/bin/pip3 /usr/bin/pip
######################
# Create Environment #
######################
# RUN cd /opt
# RUN virtualenv --python=python3.8 env
# RUN source env/bin/activate
########
# NODE #
########
RUN curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.0/install.sh | bash
RUN . "$NVM_DIR/nvm.sh" && nvm install ${NODE_VERSION}
RUN . "$NVM_DIR/nvm.sh" && nvm use v${NODE_VERSION}
RUN . "$NVM_DIR/nvm.sh" && nvm alias default v${NODE_VERSION}
RUN node --version
RUN npm --version
#########
# Torch #
#########
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
RUN python -m torch.utils.collect_env
##############
# Clone Repo #
##############
#RUN git clone https://github.com/cocktailpeanut/dalai.git /opt/dalai
RUN git clone https://github.com/oobabooga/text-generation-webui
RUN cd /opt/text-generation-webui
# RUN npm install
RUN python -m pip install -r requirements.txt
#####################
# install Alpaca 7B #
#####################
RUN npx dalai alpaca install 7B
#RUN npx dalai serve