2. 在 Python 程序中设置 如果你希望在 Python 脚本中设置该环境变量,可以使用以下代码: importos os.environ['PYTORCH_CUDA_ALLOC_CONF']='max_split_size_mb:128' 1. 2. 3. 这将确保在当前 Python 进程中PYTORCH_CUDA_ALLOC_CONF变量被设置。 示例代码 下面是一个使用 PyTorch 的示例代码,再加上设置PYTOR...