Spaces:
Sleeping
Sleeping
File size: 1,185 Bytes
0245944 b6f1234 0245944 b6f1234 0245944 |
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 |
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
RUN apt-get update && apt-get install -y libxrender1 build-essential
RUN pip install torch-sparse -f https://data.pyg.org/whl/torch-1.13.1+cu116.html
RUN pip install torch-scatter==2.0.9 torch-cluster==1.6.0 torch-spline-conv==1.2.1 torch-geometric==2.1.0 -f https://data.pyg.org/whl/torch-1.13.1+cu116.html
# Create a new user named "user" with UID 1000
RUN useradd -m -u 1000 user
# Set environment variables
ENV PYTHONUNBUFFERED=1 \
GRADIO_ALLOW_FLAGGING=never \
GRADIO_NUM_PORTS=1 \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_THEME=huggingface \
SYSTEM=spaces \
HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Set the working directory to the user's app directory as root
WORKDIR $HOME/app
# Copy the current directory contents into the container at $HOME/app
COPY . $HOME/app
# Change ownership of the app directory to "user"
RUN chown -R user:user $HOME/app
# Switch to the "user" user
USER user
# Upgrade pip as the user
RUN pip install --no-cache-dir --upgrade pip
# Install the local package as the user
RUN pip install --user .
# Set the default command to bash
CMD ["python", "app.py"] |