|
|
@ -37,7 +37,7 @@ RUN cp -ar /src /app
|
|
|
|
# Base
|
|
|
|
# Base
|
|
|
|
FROM app_base AS app_nvidia
|
|
|
|
FROM app_base AS app_nvidia
|
|
|
|
# Install pytorch for CUDA 12.1
|
|
|
|
# Install pytorch for CUDA 12.1
|
|
|
|
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \
|
|
|
|
RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
|
|
|
|
--index-url https://download.pytorch.org/whl/cu121
|
|
|
|
--index-url https://download.pytorch.org/whl/cu121
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
RUN pip3 install -r /app/requirements.txt
|
|
|
|
RUN pip3 install -r /app/requirements.txt
|
|
|
@ -53,7 +53,7 @@ RUN chmod +x /scripts/build_extensions.sh && \
|
|
|
|
# Base
|
|
|
|
# Base
|
|
|
|
FROM app_base AS app_rocm
|
|
|
|
FROM app_base AS app_rocm
|
|
|
|
# Install pytorch for ROCM
|
|
|
|
# Install pytorch for ROCM
|
|
|
|
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \
|
|
|
|
RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
|
|
|
|
--index-url https://download.pytorch.org/whl/rocm5.6
|
|
|
|
--index-url https://download.pytorch.org/whl/rocm5.6
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
RUN pip3 install -r /app/requirements_amd.txt
|
|
|
|
RUN pip3 install -r /app/requirements_amd.txt
|
|
|
@ -89,7 +89,7 @@ RUN chmod +x /scripts/build_extensions.sh && \
|
|
|
|
# Base
|
|
|
|
# Base
|
|
|
|
FROM app_base AS app_cpu
|
|
|
|
FROM app_base AS app_cpu
|
|
|
|
# Install pytorch for CPU
|
|
|
|
# Install pytorch for CPU
|
|
|
|
RUN pip3 install torch==2.1.* torchvision==0.16.* torchaudio==2.1.* \
|
|
|
|
RUN pip3 install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 \
|
|
|
|
--index-url https://download.pytorch.org/whl/cpu
|
|
|
|
--index-url https://download.pytorch.org/whl/cpu
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
# Install oobabooga/text-generation-webui
|
|
|
|
RUN pip3 install -r /app/requirements_cpu_only.txt
|
|
|
|
RUN pip3 install -r /app/requirements_cpu_only.txt
|
|
|
|