Roger Peng
Results
1
comments of
Roger Peng
if torch.cuda.is_available() and not args.cpu: args.device = "cuda" else: args.device = "cpu" CPU利用率很高而GPU利用率几乎没有变化，感觉应该是没有充分利用到GPU的CUDA来加速，不知道是不是GPU的CUDA驱动程序的版本问题。