From 869e1c86d80dee01b39d447e741f9498d917a418 Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Wed, 9 Dec 2015 11:24:07 +0100
Subject: cuda

---
 cuda/meminfo.cu | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 cuda/meminfo.cu

(limited to 'cuda/meminfo.cu')

diff --git a/cuda/meminfo.cu b/cuda/meminfo.cu
new file mode 100644
index 0000000..aebe709
--- /dev/null
+++ b/cuda/meminfo.cu
@@ -0,0 +1,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;
+}
+
-- 
cgit v1.2.3