run with nvidia gpus (#169)

huan · Apr 17, 2022 · ba83bb4 · ba83bb4
1 parent cdeedbc
commit ba83bb4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.14.4
+0.15.0
diff --git a/dochat.sh b/dochat.sh
@@ -79,6 +79,9 @@ function main () {
   for DEVICE in /dev/video* /dev/snd; do
     DEVICE_ARG+=('--device' "$DEVICE")
   done
+  if [[ $(lshw -C display | grep vendor) =~ NVIDIA ]]; then
+    DEVICE_ARG+=('--gpus' 'all' '--env' 'NVIDIA_DRIVER_CAPABILITIES=all')
+  fi
 
   echo '🚀 Starting DoChat /dɑɑˈtʃæt/ ...'
   echo

diff --git a/scripts/run.sh b/scripts/run.sh
@@ -6,10 +6,13 @@ set -x
 OPTIONS=()
 
 if [ -f /dev/snd ]; then
-  OPTIONS+=('--device /dev/snd')
+  OPTIONS+=('--device' '/dev/snd')
 fi
 if [ -f /dev/video0 ]; then
-  OPTIONS+=('--device /dev/video0')
+  OPTIONS+=('--device' '/dev/video0')
+fi
+if [[ $(lshw -C display | grep vendor) =~ NVIDIA ]]; then
+  OPTIONS+=('--gpus' 'all' '--env' 'NVIDIA_DRIVER_CAPABILITIES=all')
 fi
 
 docker run \