os.environ环境变量以及GPU的指定以及内存占用