目录:
Get torch computation device automatically based on GPU availability and memory usage
device
device – Computation device