Solved by Mona_Jalal in post #13 
              
              
                In CentOS 7, installed Python 3.8.5 and installed the version of PyTorch 1.8.1 that works with CUDA 10.2, and the problem is resolved. Running with batch size 128 below. 
(phosa) [jalal@goku phosa]$ python demo.py --filename input/bat_sidehold.jpg --class_name bat
2021-03-31 03:41:53,935 INFO     Ca…