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 |
|