Commit
•
4b0624a
1
Parent(s):
11550c8
Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
|
|
|
|
|
|
| 2 |
ARG MODEL_PART_A
|
| 3 |
ARG MODEL_PART_B
|
| 4 |
ARG MODEL_NAME
|
|
|
|
| 1 |
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
| 2 |
+
RUN apt update && apt install git build-essential libopenblas-dev wget python3-pip -y
|
| 3 |
+
RUN apt update && apt install -y wget
|
| 4 |
ARG MODEL_PART_A
|
| 5 |
ARG MODEL_PART_B
|
| 6 |
ARG MODEL_NAME
|