LocalPreferences.toml file with the desired GPU backend.
If backend == "", then the gpu_backend preference is deleted. Otherwise, backend is validated to be one of the possible backends and the preference is set to backend.
If a new backend is successfully set, then the Julia session must be restarted for the change to take effect.
- +cpu_device() -> LuxCPUDevice()
Return a LuxCPUDevice object which can be used to transfer data to CPU.
#
LuxDeviceUtils.gpu_device — Function.
gpu_device(; force_gpu_usage::Bool=false) -> AbstractLuxDevice()
@@ -498,7 +498,7 @@ Data TransferIf no GPU device is functional and force_gpu_usage is false, then cpu_device() is invoked.
If nothing works, an error is thrown.
-
+
Miscellaneous¤
@@ -507,7 +507,7 @@ Miscellaneousreset_gpu_device!()
Resets the selected GPU device. This is useful when automatic GPU selection needs to be run again.
- +#
LuxDeviceUtils.supported_gpu_backends — Function.
supported_gpu_backends() -> Tuple{String, ...}
@@ -517,7 +517,7 @@ MiscellaneousWarning
This is not the list of functional backends on the system, but rather backends which Lux.jl supports.