diff --git a/gprMax/updates.py b/gprMax/updates.py index 94722557..af6060cc 100644 --- a/gprMax/updates.py +++ b/gprMax/updates.py @@ -259,7 +259,7 @@ class CUDAUpdates: self.drv.init() # Create device handle and context on specific GPU device (and make it current context) - self.dev = self.drv.Device(config.get_model_config().cuda['gpu'].deviceID) + self.dev = config.get_model_config().device['dev'] self.ctx = self.dev.make_context() # Initialise arrays on GPU, prepare kernels, and get kernel functions diff --git a/gprMax/utilities/host_info.py b/gprMax/utilities/host_info.py index 0e860fb4..d81486bc 100644 --- a/gprMax/utilities/host_info.py +++ b/gprMax/utilities/host_info.py @@ -428,7 +428,7 @@ def print_cuda_info(devs): logger.debug(f'PyCUDA: {pycuda.VERSION_TEXT}') for ID, gpu in devs.items(): - logger.basic(f" |--->Device {ID}: {' '.join(gpu.name.split())} | " + logger.basic(f" |--->Device {ID}: {' '.join(gpu.name().split())} | " f"{human_size(gpu.total_memory(), a_kilobyte_is_1024_bytes=True)}")