diff --git a/tinygrad/device.py b/tinygrad/device.py index f8f18b1777..dd42cc8992 100644 --- a/tinygrad/device.py +++ b/tinygrad/device.py @@ -29,7 +29,7 @@ class _Device: def DEFAULT(self) -> str: device_from_env: Optional[str] = functools.reduce(lambda val, ele: ele if getenv(ele) == 1 else val, self._devices, None) # type: ignore if device_from_env: return device_from_env - for device in ["METAL", "AMD", "CUDA", "GPU", "CLANG", "LLVM"]: + for device in ["METAL", "AMD", "NV", "CUDA", "GPU", "CLANG", "LLVM"]: try: if self[device]: os.environ[device] = "1" # we set this in environment for spawned children