Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +7 -1
Dockerfile
CHANGED
|
@@ -1,5 +1,11 @@
|
|
| 1 |
# Use an official Python runtime as a parent image
|
| 2 |
-
FROM
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
|
| 4 |
# Set up a new user named "user" with user ID 1000, Essential
|
| 5 |
RUN useradd -m -u 1000 user
|
|
|
|
| 1 |
# Use an official Python runtime as a parent image
|
| 2 |
+
FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
|
| 3 |
+
|
| 4 |
+
# Set up environment variable, Important to use gpu in the container
|
| 5 |
+
ENV NVIDIA_VISIBLE_DEVICES all
|
| 6 |
+
ENV NVIDIA_DRIVER_CAPABILITIES all
|
| 7 |
+
|
| 8 |
+
# FROM python:3.12.3
|
| 9 |
|
| 10 |
# Set up a new user named "user" with user ID 1000, Essential
|
| 11 |
RUN useradd -m -u 1000 user
|