#include 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; }