1 2 3 4 5 6
#include <cuda.h> #include <cuda_runtime.h> #include <cudnn.h> #include <cublas_v2.h> #include <cublasLt.h> #include <nvrtc.h>