Browse Source

demo nb run with cuda

devel
Stefan Holst 4 years ago
parent
commit
5084f1dd8c
  1. 40
      Demo.ipynb

40
Demo.ipynb

@ -18,15 +18,7 @@
"cell_type": "code", "cell_type": "code",
"execution_count": 1, "execution_count": 1,
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [],
{
"name": "stdout",
"output_type": "stream",
"text": [
"0000000.000 W Numba unavailable. Falling back to pure Python.\n"
]
}
],
"source": [ "source": [
"from kyupy import bench\n", "from kyupy import bench\n",
"\n", "\n",
@ -1291,15 +1283,27 @@
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
"ename": "ModuleNotFoundError", "name": "stdout",
"evalue": "No module named 'numba'", "output_type": "stream",
"output_type": "error", "text": [
"traceback": [ "Found 1 CUDA devices\n",
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m", "id 0 b'TITAN V' [SUPPORTED]\n",
"\u001b[0;31mModuleNotFoundError\u001b[0m Traceback (most recent call last)", " compute capability: 7.0\n",
"\u001b[0;32m<ipython-input-45-ffb50e0958fc>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0;32mfrom\u001b[0m \u001b[0mnumba\u001b[0m \u001b[0;32mimport\u001b[0m \u001b[0mcuda\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 2\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 3\u001b[0m \u001b[0mcuda\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdetect\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n", " pci device id: 0\n",
"\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'numba'" " pci bus id: 2\n",
"Summary:\n",
"\t1/1 devices are supported\n"
] ]
},
{
"data": {
"text/plain": [
"True"
]
},
"execution_count": 45,
"metadata": {},
"output_type": "execute_result"
} }
], ],
"source": [ "source": [
@ -1332,7 +1336,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.9.1" "version": "3.6.8"
} }
}, },
"nbformat": 4, "nbformat": 4,

Loading…
Cancel
Save