mt_gpu.py

# -*- coding: utf-8 -*-

import cupy
import threading
import random
import sys

# enable memory pool
cupy.cuda.memory.set_allocator(cupy.cuda.memory.MemoryPool().malloc)

def run():
    cupy.cuda.runtime.free(0)
    while True:
        print(".")
        x = cupy.arange(int(2000000 * random.random()))
        for i in range(10):
            x = x - 1

threads = []
for x in range(2):
    threads.append(threading.Thread(target=run))

for t in threads: t.setDaemon(True)
for t in threads: t.start()
for t in threads: t.join()