kromskii2/alltalk_tts
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
erew123 authored and GitHub committed 2024-04-04 10:58:35 +01:00
1 parent aa7ff1940a
commit a96c82ff01
1 file changed
+6 -2
Download Patch File Download Diff File Expand all files Collapse all files
nvidia.Dockerfile
+6 -2
View File
@@ -7,12 +7,16 @@ RUN apt-get update && apt-get upgrade -y \
libclblast-dev libopenblas-dev \ libclblast-dev libopenblas-dev \
&& mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd \ && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd \
&& ln -s /usr/bin/python3 /usr/bin/python && ln -s /usr/bin/python3 /usr/bin/python
RUN mkdir /app/pip_cache
ENV PIP_CACHE_DIR=/app/pip_cache
WORKDIR /app WORKDIR /app
COPY . . COPY . .
ENV CUDA_DOCKER_ARCH=all ENV CUDA_DOCKER_ARCH=all
RUN pip install --no-cache-dir -r system/requirements/requirements_standalone.txt --cache-dir=/app/pip_cache && \
pip install --no-cache-dir deepspeed --cache-dir=/app/pip_cache
RUN pip install --no-cache-dir --no-deps -r system/requirements/requirements_standalone.txt && \
pip install --no-cache-dir deepspeed
EXPOSE 7851 EXPOSE 7851
RUN chmod +x launch.sh RUN chmod +x launch.sh
ENTRYPOINT ["sh", "-c", "./launch.sh"] ENTRYPOINT ["sh", "-c", "./launch.sh"]