File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -2,15 +2,16 @@ version: "3.8"
2
2
# Compose file build variables set in .env
3
3
services :
4
4
supervisor :
5
+ platform : linux/amd64
5
6
build :
6
7
context : ./build
7
8
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}
9
10
PYTORCH_VERSION : ${PYTORCH_VERSION:-2.1.1}
10
11
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}"
12
13
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}
14
15
15
16
# # For Nvidia GPU's - You probably want to uncomment this
16
17
# deploy:
You can’t perform that action at this time.
0 commit comments