gpu fix

2023-06-17 09:33:04 +00:00
parent 422bdbfcd5
commit 2c873b255e
1 changed files with 2 additions and 2 deletions
@@ -1,8 +1,8 @@
 import os
 import sys

-os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"] = '0'
+# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+# os.environ["CUDA_VISIBLE_DEVICES"] = '0'

 import torch
 from torch import nn, Tensor