用终端命令释放 PyTorch 占用的内存。
时间: 2024-02-13 22:07:53 浏览: 211
Pytorch释放显存占用方式
你可以使用以下命令来释放 PyTorch 占用的 GPU 内存:
```
$ fuser -v /dev/nvidia* # 查看哪些进程在使用 GPU
$ kill -9 <PID> # 结束对应的进程
$ sudo fuser -v /dev/nvidia* # 确认没有进程在使用 GPU
$ sudo systemctl restart nvidia-persistenced.service # 重启 NVIDIA Persistence Daemon
$ sudo systemctl restart nvidia-smi.service # 重启 NVIDIA System Management Interface
```
其中,`<PID>` 是占用 GPU 内存的进程的 PID。你可以使用 `fuser -v /dev/nvidia*` 命令来查看哪些进程在使用 GPU。如果你想要释放 PyTorch 占用的内存,你需要找到 PyTorch 进程的 PID,并使用 `kill -9 <PID>` 命令来结束这个进程。注意,这个操作会结束整个进程,如果你有其他任务在这个进程中运行,这些任务也会被结束。另外,这个方法只能释放 PyTorch 占用的 GPU 内存,不能释放 CPU 内存。如果你想要释放 CPU 内存,你可以使用 Linux 内置的 `free` 命令来查看内存使用情况,并使用 `sudo sync && sudo sysctl -w vm.drop_caches=3` 命令来释放缓存。
阅读全文