python查看进程占用显存

可以使用`psutil`库来查看Python进程占用的显存。

以下是一个简单的示例代码：

import psutil

pid = <your_process_id>

process = psutil.Process(pid)

gpu_memory = process.memory_info().vms / (1024 * 1024)

print(f"Process {pid} is using {gpu_memory:.2f} MB of virtual memory")

其中，`<your_process_id>`需要替换为你要查看的Python进程的PID，`psutil.Process()`方法可以根据PID获取进程对象，然后通过`memory_info()`方法获取进程的内存信息，其中`vms`属性代表进程占用的虚拟内存大小，单位为字节，需要将其转换为MB来显示。