Spaces:
Running
Running
kiran kumar commited on
Commit ·
c700583
1
Parent(s): 5d988d2
Initial Dockerized Flask VAE backend
Browse files- Dockerfile +0 -2
- requirements.txt +0 -0
Dockerfile
CHANGED
|
@@ -1,9 +1,7 @@
|
|
| 1 |
# Use official PyTorch image with CUDA 11.8 and cuDNN 9
|
| 2 |
FROM pytorch/pytorch:2.7.1-cuda11.8-cudnn9-runtime
|
| 3 |
-
|
| 4 |
# Set working directory inside the container
|
| 5 |
WORKDIR /app
|
| 6 |
-
|
| 7 |
# Copy requirements.txt first (for layer caching)
|
| 8 |
COPY requirements.txt .
|
| 9 |
|
|
|
|
| 1 |
# Use official PyTorch image with CUDA 11.8 and cuDNN 9
|
| 2 |
FROM pytorch/pytorch:2.7.1-cuda11.8-cudnn9-runtime
|
|
|
|
| 3 |
# Set working directory inside the container
|
| 4 |
WORKDIR /app
|
|
|
|
| 5 |
# Copy requirements.txt first (for layer caching)
|
| 6 |
COPY requirements.txt .
|
| 7 |
|
requirements.txt
CHANGED
|
Binary files a/requirements.txt and b/requirements.txt differ
|
|
|