Skip to content

Latest commit

 

History

History
5 lines (3 loc) · 312 Bytes

pytorch.md

File metadata and controls

5 lines (3 loc) · 312 Bytes

PyTorch踩坑手记

  • 多GPU模式下,不能和其他进程共享同一个GPU,否则会出现ConnectionError,应该用环境变量将自己的进程使用的GPU和其他进程使用的GPU分开

  • 循环中定义的变量,如果在循环结束后不需要使用,需要del,避免一直占用显存