From 08e836108f1a6765ba04658be9b5172921a1e03a Mon Sep 17 00:00:00 2001
From: Jonas Kulhanek <jonas.kulhanek@live.com>
Date: Wed, 14 Aug 2024 20:09:52 +0200
Subject: [PATCH] Drop set_cuda_device

---
 nerfstudio/scripts/train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/nerfstudio/scripts/train.py b/nerfstudio/scripts/train.py
index 09cee86aa6..fde0a5c37a 100644
--- a/nerfstudio/scripts/train.py
+++ b/nerfstudio/scripts/train.py
@@ -95,8 +95,6 @@ def train_loop(local_rank: int, world_size: int, config: TrainerConfig, global_r
         config: config file specifying training regimen
     """
     _set_random_seed(config.machine.seed + global_rank)
-    if torch.cuda.is_available():
-        torch.cuda.set_device(local_rank)
     trainer = config.setup(local_rank=local_rank, world_size=world_size)
     trainer.setup()
     trainer.train()