| FROM dnth/ram-plus-image-tagging-gpu | |
| # Set BentoML home to a writable directory | |
| ENV BENTOML_HOME=/home/user/bentoml | |
| # Create a non-root user | |
| RUN useradd -m -u 1000 user | |
| # Create BentoML home directory and set permissions | |
| RUN mkdir -p /home/user/bentoml && chown -R user:user /home/user/bentoml | |
| # Switch to non-root user | |
| USER user |