Skip to content

Instantly share code, notes, and snippets.

@jakirkham
jakirkham / pynvml_query_memory.py
Created January 13, 2020 22:50 — forked from pentschev/pynvml_query_memory.py
Query used GPU memory with pynvml
import datetime
import getopt
import os
import sys
import time
import pynvml
def print_mem(dev_count):
for index in range(dev_count):
@jakirkham
jakirkham / dask_cudf_merge_benchmark.ipynb
Created December 3, 2019 16:19 — forked from rjzamora/dask_cudf_merge_benchmark.ipynb
dask_cudf + UCX merge benchmark
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@jakirkham
jakirkham / dask_image_watershed.ipynb
Created November 12, 2019 21:17 — forked from wwarriner/dask_image_watershed.ipynb
First attempt at distributed watershed
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@jakirkham
jakirkham / conda_pkg_urls_to_whls.py
Last active November 9, 2019 03:09
Searches for matching Conda packages, downloads them, and converts them to wheels
#!/usr/bin/env python
import argparse
import contextlib
import json
import os
import shutil
import tempfile
import subprocess
@jakirkham
jakirkham / ridge-cuml-cudf-sklearn.ipynb
Created October 9, 2019 23:51
Simple notebook using scikit-learn's GridSearch with cuML and cuDF
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@jakirkham
jakirkham / ridge-cuml-cupy-sklearn.ipynb
Created September 12, 2019 18:10
Simple notebook using scikit-learn's GridSearch with cuML and CuPy
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@jakirkham
jakirkham / cupy_chain_conv_test.ipynb
Created September 11, 2019 17:18
Attempt at using CuPy + Chainer to mimic SciPy's convolve
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@jakirkham
jakirkham / array_indexing.ipynb
Last active August 21, 2019 22:42
Simple benchmark of indexing NumPy and CuPy arrays
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
import asyncio
import time
import numpy as np
import cupy
import numba
import dask.array as da
from dask_cuda import DGX, LocalCUDACluster
from dask.distributed import Client, wait