Skip to content

Commit 18421a9

Browse files
committed
add platform, default to runtime base [skip ci]
1 parent 53cfd35 commit 18421a9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

docker-compose.yaml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,16 @@ version: "3.8"
22
# Compose file build variables set in .env
33
services:
44
supervisor:
5+
platform: linux/amd64
56
build:
67
context: ./build
78
args:
8-
IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:3.10-cuda-11.8.0-base-22.04}
9+
IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:3.10-cuda-11.8.0-runtime-22.04}
910
PYTORCH_VERSION: ${PYTORCH_VERSION:-2.1.1}
1011
tags:
11-
- "ghcr.io/ai-dock/pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-base-22.04}"
12+
- "ghcr.io/ai-dock/pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-runtime-22.04}"
1213

13-
image: ghcr.io/ai-dock/pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-base-22.04}
14+
image: ghcr.io/ai-dock/pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-runtime-22.04}
1415

1516
## For Nvidia GPU's - You probably want to uncomment this
1617
#deploy:

0 commit comments

Comments
 (0)