diff --git a/dask_cuda/local_cuda_cluster.py b/dask_cuda/local_cuda_cluster.py index 0329e658c..4a706e67f 100644 --- a/dask_cuda/local_cuda_cluster.py +++ b/dask_cuda/local_cuda_cluster.py @@ -309,7 +309,7 @@ def __init__( elif ucx_net_devices == "": raise ValueError("ucx_net_devices can not be an empty string") self.ucx_net_devices = ucx_net_devices - self.set_ucx_net_devices = enable_infiniband + self.set_ucx_net_devices = enable_infiniband and ucx_net_devices is not None self.host = kwargs.get("host", None) initialize(