CUDA_CUDA_VISIBLE_DEVICES

PyTorch通过设置PYTORCH_CUDA_ALLOC_CONF环境变量中的max_split_size_mb参数来优化CUDA显存分配，从而缓解CUDA Out of Memory错误。