Spaces:
Sleeping
Sleeping
Commit
•
a197417
1
Parent(s):
3d66ec5
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM
|
| 2 |
WORKDIR /usr/src/app
|
| 3 |
COPY requirements.txt ./
|
| 4 |
RUN python -m pip install --upgrade pip setuptools wheel
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.3.1-base-ubuntu20.04
|
| 2 |
WORKDIR /usr/src/app
|
| 3 |
COPY requirements.txt ./
|
| 4 |
RUN python -m pip install --upgrade pip setuptools wheel
|