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
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import datetime | |
| import getopt | |
| import os | |
| import sys | |
| import time | |
| import pynvml | |
| def print_mem(dev_count): | |
| for index in range(dev_count): |
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
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ dask_image_watershed.ipynb
Created
November 12, 2019 21:17
— forked from wwarriner/dask_image_watershed.ipynb
First attempt at distributed watershed
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
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
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python | |
| import argparse | |
| import contextlib | |
| import json | |
| import os | |
| import shutil | |
| import tempfile | |
| import subprocess |
jakirkham
/ ridge-cuml-cudf-sklearn.ipynb
Created
October 9, 2019 23:51
Simple notebook using scikit-learn's GridSearch with cuML and cuDF
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ ridge-cuml-cupy-sklearn.ipynb
Created
September 12, 2019 18:10
Simple notebook using scikit-learn's GridSearch with cuML and CuPy
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ cupy_chain_conv_test.ipynb
Created
September 11, 2019 17:18
Attempt at using CuPy + Chainer to mimic SciPy's convolve
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ cudf-iloc-timings.ipynb
Created
August 21, 2019 23:50
— forked from quasiben/cudf-iloc-timings.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ array_indexing.ipynb
Last active
August 21, 2019 22:42
Simple benchmark of indexing NumPy and CuPy arrays
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
jakirkham
/ ucx_map_overlap_smooth.py
Created
August 16, 2019 10:06
— forked from pentschev/ucx_map_overlap_smooth.py
UCX map overlap sample
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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 |