blob: aebe709d0072fdae8acbc6bf52087ad270509057 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
#include <iostream>
using namespace std;
int
main(void)
{
size_t free_bytes;
size_t total_bytes;
cudaSetDevice(0);
cudaMemGetInfo( &free_bytes, &total_bytes );
cerr << "[cnn] Memory Free (MB): " << free_bytes/1024/1024 << "/" << total_bytes/1024/1024 << endl << endl;
}
|