update fix

2026-09-24 13:00:12 +08:00 · 2019-09-18 23:02:03 -06:00
parent 23c5c0bd9e
commit 1db6b7a319
2 changed files with 2 additions and 1 deletions
@@ -20,6 +20,7 @@ cd ..
 curl https://sdk.cloud.google.com | bash
 exec -l $SHELL
 gcloud init
+export CUDA_VISIBLE_DEVICES=0

 # Download the 512-length model if specified, 256-length otherwise
 #if [ "$1" = "512" ]