fix type mismatch while call torch._C._cuda_setDevice (pytorch#8065)

* fix type mismatch while call torch._C._cuda_setDevice * fix type mismatch in scatter * fix type mismatch in scatter * fix type mismatch while call torch._C._cuda_setDevice * fix type mismatch while call torch._C._cuda_setDevice * fix type mismatch while call torch._C._cuda_setDevice
flyingcarpet-network · Jun 5, 2018 · f5cd479 · f5cd479
1 parent c446269
commit f5cd479
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/cuda/__init__.py b/torch/cuda/__init__.py
@@ -216,7 +216,7 @@ class device(object):
     """
 
     def __init__(self, idx):
-        self.idx = idx
+        self.idx = int(idx)
         self.prev_idx = -1
 
     def __enter__(self):