image env

Author: Corentin

myoquant-sdh-train.ipynb

@@ -0,0 +1,58 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'\n",
+    "\n",
+    "import tensorflow as tf\n",
+    "if tf.test.gpu_device_name()=='':\n",
+    "  print('You do not have GPU access.') \n",
+    "  print('Did you change your runtime ?') \n",
+    "  print('If the runtime setting is correct then Google did not allocate a GPU for your session')\n",
+    "  print('Expect slow performance. To access GPU try reconnecting later')\n",
+    "\n",
+    "else:\n",
+    "  print('You have GPU access')\n",
+    "  !nvidia-smi\n",
+    "\n",
+    "# from tensorflow.python.client import device_lib \n",
+    "# device_lib.list_local_devices()\n",
+    "\n",
+    "# print the tensorflow version\n",
+    "print('Tensorflow version is ' + str(tf.__version__))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.6"
+  },
+  "orig_nbformat": 4,
+  "vscode": {
+   "interpreter": {
+    "hash": "185732d1a0c8de10c9bb03975f56b22bfc308b23770b008cd7b656c75bc57aea"
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}