Add K80 templates
@@ -29,6 +29,12 @@ elif [[ $INSTANCE_TYPE == "gpu-2" ]]; then
TEMPLATE="gpu-worker-template-2"
elif [[ $INSTANCE_TYPE == "gpu-4" ]]; then
TEMPLATE="gpu-worker-template-4"
+elif [[ $INSTANCE_TYPE == "gpu-k1" ]]; then
+ TEMPLATE="gpu-worker-template-k1"
+elif [[ $INSTANCE_TYPE == "gpu-k2" ]]; then
+ TEMPLATE="gpu-worker-template-k2"
+elif [[ $INSTANCE_TYPE == "gpu-k4" ]]; then
+ TEMPLATE="gpu-worker-template-k4"
else
echo "Unsupported instance type '$INSTANCE_TYPE'"
exit 1
