...

gradio_demo/barc_demo_v3.py

@@ -4,8 +4,8 @@ import os
 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")
-    os.system("pip install --upgrade  torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html")
+    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")
+    # os.system("pip install --upgrade  torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html")
 except Exception as e:
     print(e)
 
@@ -33,6 +33,12 @@ from combined_model.model_shape_v7 import ModelImageTo3d_withshape_withproj
 
 from configs.barc_cfg_defaults import get_cfg_global_updated
 
+print(
+    "torch: ", torch.__version__,
+    "\ntorchvision: ", torchvision.__version__,
+)
+print("EnV", os.environ)
+
 
 
 def get_prediction(model, img_path_or_img, confidence=0.5):