ERROR -->
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-21-2620be06dd37> in <module>
      4 prev_time = time.time()
      5 img = Image.open(img_path)
----> 6 detections = detect_image(img)
      7 inference_time = datetime.timedelta(seconds=time.time() - prev_time)
      8 #print ('Inference Time: %s' % (inference_time))

<ipython-input-20-5e3b41feb37a> in detect_image(img)
     17     image_tensor = image_tensor.unsqueeze_(0)
     18     print(image_tensor)
---> 19     input_img = Variable(image_tensor.type(Tensor))
     20     print("3")
     21     # run inference on the model and get detections

RuntimeError: Cannot initialize CUDA without
 ATen_cuda library. PyTorch splits its backend
 into two shared libraries: a CPU library and a 
CUDA library; this error has occurred because
 you are trying to use some CUDA functionality,
 but the CUDA library has not been loaded by 
the dynamic linker for some reason.  The CUDA
 library MUST be loaded, EVEN IF you don't 
directly use any symbols from the CUDA library! 
One common culprit is a lack of -Wl,--no-as-
needed in your link arguments; many dynamic 
linkers will delete dynamic library dependencies 
if you don't depend on any of their symbols.  
You can check if this has occurred by using ldd 
on your binary to see if there is a dependency 
on *_cuda.so library.