1 #include "cuda_runtime.h"
 2 #include "device_launch_parameters.h"
 3 #include <iostream>
 4 using namespace std;
 5 int main()
 6 {
 7     int *dev_buf = 0;
 8     cudaError_t cudaStatus;
 9     cudaStatus = cudaSetDevice(0);//选择GPU
10     if (cudaStatus != cudaSuccess) 
11     {
12         cout << "GPU设置错误\n";
13         goto Error;
14     }
15     cudaStatus = cudaMalloc((void**)&dev_buf, 5 * sizeof(int));//申请空间
16     cudaStatus = cudaDeviceReset();//结束，GPU重置
17     if (cudaStatus != cudaSuccess) 
18     {
19         cout<<"GPU重置错误";
20         return 1;
21     }
22 Error:
23     cudaFree(dev_buf);
24     return 0;
25 }