Update PIFu/spaces.py

PIFu/spaces.py

@@ -1,5 +1,6 @@
 import os
-from xml.etree.ElementPath import ops
+os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"  
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
 try:
     os.system("pip install --upgrade  torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html")
 except Exception as e:
@@ -23,8 +24,6 @@ from types import SimpleNamespace
 import trimesh
 import glob
 
-os.environ["CUDA_VISIBLE_DEVICES"]="0"
-
 print(
     "torch: ", torch.__version__,
     "\ntorchvision: ", torchvision.__version__,