Skip to content

Commit

Permalink
Remove TPU Availability check from parse devices
Browse files Browse the repository at this point in the history
  • Loading branch information
kaushikb11 committed Mar 14, 2022
1 parent 7ee6907 commit 1764f63
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions pytorch_lightning/utilities/device_parser.py
Expand Up @@ -17,7 +17,6 @@

from pytorch_lightning.plugins.environments import TorchElasticEnvironment
from pytorch_lightning.tuner.auto_gpu_select import pick_multiple_gpus
from pytorch_lightning.utilities import _TPU_AVAILABLE
from pytorch_lightning.utilities.exceptions import MisconfigurationException
from pytorch_lightning.utilities.types import _DEVICE

Expand Down Expand Up @@ -131,8 +130,8 @@ def parse_tpu_cores(tpu_cores: Optional[Union[int, str, List[int]]]) -> Optional
if not _tpu_cores_valid(tpu_cores):
raise MisconfigurationException("`tpu_cores` can only be 1, 8 or [<1-8>]")

if tpu_cores is not None and not _TPU_AVAILABLE:
raise MisconfigurationException("No TPU devices were found.")
if tpu_cores is not None:
raise MisconfigurationException(f"No TPU devices for {tpu_cores} were found.")

return tpu_cores

Expand Down

0 comments on commit 1764f63

Please sign in to comment.