The total number of bytes of memory this GPU has.
()
| 55 | |
| 56 | |
| 57 | def device_mem_total() -> int: |
| 58 | """The total number of bytes of memory this GPU has.""" |
| 59 | import cuda.bindings.runtime as cudart |
| 60 | |
| 61 | status, _, total = cudart.cudaMemGetInfo() |
| 62 | _checkcu(status) |
| 63 | return total |
| 64 | |
| 65 | |
| 66 | def make_batches( |
no test coverage detected