torch的torch device 有哪些mps是哪种
torch device 有 "cpu"、"cuda"、"mkldnn"、"opengl"、"opencl"、"ideep"、"hip"、"msnpu"、"xla" 等。其中,"cuda" 是基于 NVIDIA GPU 的加速器,"cpu" 是基于 CPU 的计算设备。
MPS 是 CUDA 中的一种多进程协作技术,用于在一个 GPU 上同时执行多个 CUDA 进程,可以提高 GPU 的利用率和并行性能。MPS 不算是一种 torch device,而是 CUDA 原生支持的一种多任务管理技术。
原文地址: https://www.cveoy.top/t/topic/clqG 著作权归作者所有。请勿转载和采集!