程序员经验分享-hwhale

  • Pytorch GPU编号问题

    虽然os environ CUDA VISIBLE DEVICES 能够实现在代码中设置可见的GPU 设备 但是在实际运行中 Pytorch会对这些GPU重新编号 比如 os environ CUDA VISIBLE DEVICES 3 5
    Pytorch GPU编号