python清除子进程占用显存

在 Python 中，可以使用 `multiprocessing` 模块来创建子进程。要清除子进程占用的显存，可以使用以下代码：

import psutil
import os
import multiprocessing as mp

def clear_memory():
    pid = os.getpid()
    py = psutil.Process(pid)
    memory_use = py.memory_info()[0]/2.**30
    print("Memory use:", memory_use)
    mp.current_process()._config['tempdir'] = '/dev/shm'
    mp.set_start_method('spawn', True)

if __name__ == '__main__':
    clear_memory()
    # 在这里创建子进程，执行任务

在上面的代码中，首先使用 `psutil` 模块获取当前进程的 PID，然后通过 `memory_info()` 方法获取当前进程占用的内存大小。接着，将 `tempdir` 设置为 `/dev/shm`，这样可以将临时文件存储在内存中，而不是存储在磁盘上。最后，使用 `spawn` 方法创建子进程，以确保每个进程都有自己独立的内存空间。

需要注意的是，在使用 `multiprocessing` 模块时，需要将代码放到 `if __name__ == '__main__':` 中，否则会出现多个进程同时运行的问题。