ps:师傅希望想tensorflow一样设置 GPU占用（tf.GPUOptions），但事实情况是好像pytorch根本就没有相关函数来设置（如８Ｇ显存我只用2G,来跑是否可行） 1.网上很多攻略关于pytorch 训练时显