识别并选择GPU设备
此示例显示了如何使用gpudevice
识别并选择要使用的设备。
要确定计算机中有多少个GPU设备,请使用gpudevicecount
功能。
gpudevicecount(“可用的”)
ANS = 2
当有多个设备时,第一个是默认设备。您可以使用gpudevicetable
函数以确定您是否要使用的功能。
gpudevicetable
ans =2×5桌Index Name ComputeCapability DeviceAvailable DeviceSelected _____ _____________ _________________ _______________ ______________ 1 "TITAN RTX" "7.5" true true 2 "Quadro K620" "5.0" true false
如果第一个设备是您要使用的设备,则可以继续进行。要在GPU上运行计算,请使用gpuarray
启用功能。有关更多信息,请参阅在GPU上运行MATLAB功能。
要使用另一个设备,请致电gpudevice
带有另一个设备的索引。
gpudevice(2)
ans = CUDADevice with properties: Name: 'Quadro K620' Index: 2 ComputeCapability: '5.0' 万博1manbetxSupportsDouble: 1 DriverVersion: 11 ToolkitVersion: 10.2000 MaxThreadsPerBlock: 1024 MaxShmemPerBlock: 49152 MaxThreadBlockSize: [1024 1024 64] MaxGridSize: [2.1475e+09 6553565535] SIMDWidth: 32 TotalMemory: 2.1475e+09 AvailableMemory: 1.6776e+09 MultiprocessorCount: 3 ClockRateKHz: 1124000 ComputeMode: 'Default' GPUOverlapsTransfers: 1 KernelExecutionTimeout: 1 CanMapHostMemory: 1 Device万博1manbetxSupported: 1 DeviceAvailable: 1 DeviceSelected: 1
也可以看看
gpudevice
|gpudevicecount
|gpuarray
|gpudevicetable