changed dependencies

main.py

@@ -31,6 +31,7 @@ from examples_metadata import (
 logging.basicConfig(level=logging.DEBUG)
 os.system("pip list")
 #print Cuda version
+print("Gpu Cuda Driver Version:")
 nvmlInit()
 print(pynvml.nvmlSystemGetCudaDriverVersion())
 cuda_version = torch.version.cuda
@@ -202,6 +203,8 @@ if torch.cuda.is_available():
 else:
     print("CUDA is not available. Using CPU.")
 
+torch.zeros(1).cuda()
+
 
 def generateTags():
     start = time.time()