Spaces:
Runtime error
Runtime error
Commit
·
b6e1550
1
Parent(s):
2fd019f
add git to nvidia dockerfile
Browse files- Dockerfile.nvidia +1 -1
Dockerfile.nvidia
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04 as base-img
|
| 2 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 3 |
-
RUN apt-get update && apt-get install -y python3 ffmpeg
|
| 4 |
RUN apt-get install -y python3-pip
|
| 5 |
RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
|
| 6 |
|
|
|
|
| 1 |
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04 as base-img
|
| 2 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 3 |
+
RUN apt-get update && apt-get install -y python3 ffmpeg git libxext6 libsm6 libgtk2.0-dev pkg-config libgtk2.0-dev libavcodec-dev python3-opencv libopencv-dev
|
| 4 |
RUN apt-get install -y python3-pip
|
| 5 |
RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
|
| 6 |
|