Add back ROCm container support

This adds ROCm support back as a discrete image.
This commit is contained in:
Daniel Hiltgen
2024-01-25 16:58:05 -08:00
parent 197e420a97
commit 75c44aa319
2 changed files with 21 additions and 1 deletions

View File

@ -13,3 +13,13 @@ docker build \
-f Dockerfile \
-t ollama/ollama:$VERSION \
.
docker build \
--load \
--platform=linux/amd64 \
--build-arg=VERSION \
--build-arg=GOFLAGS \
--target runtime-rocm \
-f Dockerfile \
-t ollama/ollama:$VERSION-rocm \
.