相关疑难解决方法(0)

有没有一种方法可以为特定Pytorch实例强制提供我想使用的GPU内存量的最大值？例如，我的GPU可能有12Gb可用，但是我想为特定进程分配最大4Gb。