Skip to content

Commit

Permalink
Check not macOS when looking for TPUs [release]
Browse files Browse the repository at this point in the history
  • Loading branch information
AjayP13 committed Dec 27, 2023
1 parent 666eaef commit 8e24c3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/.cluster/direct/_direct_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ if [ -z "$PROJECT_DATA" ]; then
exit 1
fi
if [ -z "$PROJECT_ACCELERATOR_TYPE" ]; then
if (env -0 | cut -z -f1 -d= | tr '\0' '\n' | grep -q "TPU_"); then
if [ "$(uname)" != "Darwin" ] && (env -0 | cut -z -f1 -d= | tr '\0' '\n' | grep -q "TPU_"); then
export PROJECT_ACCELERATOR_TYPE=tpu
export PROJECT_VISIBLE_ACCELERATOR_DEVICES=all
echo "Detected an environment with TPUs. For this reason, accelerator device dependencies will not be installed to a virtual environment." 1>&2
Expand Down

0 comments on commit 8e24c3f

Please sign in to comment.