added fixed 5
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -1,9 +1,11 @@
|
|
| 1 |
# Use a base image with Python
|
| 2 |
-
|
|
|
|
| 3 |
|
| 4 |
# Install git and other build tools
|
| 5 |
# The specific command depends on the base image's package manager.
|
| 6 |
# For Debian/Ubuntu-based images like python:3.10-slim, it's apt-get.
|
|
|
|
| 7 |
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
| 8 |
RUN python3 -m pip install --upgrade pip
|
| 9 |
|
|
|
|
| 1 |
# Use a base image with Python
|
| 2 |
+
# Use a CUDA devel image as the base
|
| 3 |
+
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
|
| 4 |
|
| 5 |
# Install git and other build tools
|
| 6 |
# The specific command depends on the base image's package manager.
|
| 7 |
# For Debian/Ubuntu-based images like python:3.10-slim, it's apt-get.
|
| 8 |
+
RUN apt-get update && apt-get install -y python3 python3-pip git
|
| 9 |
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
| 10 |
RUN python3 -m pip install --upgrade pip
|
| 11 |
|