主要内容

识别并选择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

也可以看看

|||

相关话题