Skip to content

Commit 57d1784

Browse files
authored
fix: docker gpus option "all" support
"docker --gpus=all" is a valid and mostly used command. regex updated to match this.
1 parent 1092ee9 commit 57d1784

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

run-compose.sh

+2-1
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ usage() {
8282
echo "Examples:"
8383
echo " $0 --drop"
8484
echo " $0 --enable-gpu[count=1]"
85+
echo " $0 --enable-gpu[count=all]"
8586
echo " $0 --enable-api[port=11435]"
8687
echo " $0 --enable-gpu[count=1] --enable-api[port=12345] --webui[port=3000]"
8788
echo " $0 --enable-gpu[count=1] --enable-api[port=12345] --webui[port=3000] --data[folder=./ollama-data]"
@@ -160,7 +161,7 @@ else
160161
if [[ $enable_gpu == true ]]; then
161162
# Validate and process command-line arguments
162163
if [[ -n $gpu_count ]]; then
163-
if ! [[ $gpu_count =~ ^[0-9]+$ ]]; then
164+
if ! [[ $gpu_count =~ ^([0-9]+|all)$ ]]; then
164165
echo "Invalid GPU count: $gpu_count"
165166
exit 1
166167
fi

0 commit comments

Comments
 (0)