libtorch c++调用 (九)打印tensor 信息

libtorch c++调用 (九)打印tensor信息

  • 1、打印cuda信息
  • 2、打印tensor信息
  • 3、打印cv::mat信息

1、打印cuda信息

void test_libtorch_version() {
   
	std::cout << "cuDNN : " << torch::cuda::cudnn_is_available() <&

你可能感兴趣的:(pytorch,pytorch实践,Pytorch)