georgedimitriadis / spikesorting_tsne Goto Github PK

LOADING UP THE GPU
Loading matrix time:  0.010 s
ITERATION NUMBER: 1
Making the dot products Time: 0.008 s
Summing the dot products on the GPU Time: 0.178 s
cuBLAS gemm Time: 0.174 s
     Number of sorting segments = 6
Traceback (most recent call last):
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/gpu.py", line 189, in _segment_sort_transposed_distances_get_knns
    sorting.segmented_sort(keys=keys, vals=values, segments=segments)
  File "/opt/conda/lib/python3.6/site-packages/pyculib/sorting/segsort.py", line 115, in segmented_sort
    _segmentedsort(d_keys, d_vals, d_segments, stream)
  File "/opt/conda/lib/python3.6/contextlib.py", line 88, in __exit__
    next(self.gen)
  File "/opt/conda/lib/python3.6/site-packages/pyculib/sorting/segsort.py", line 84, in _autodevice
    dptr.copy_to_host(ary, stream=stream)
  File "/opt/conda/lib/python3.6/site-packages/numba/cuda/cudadrv/devices.py", line 212, in _require_cuda_context
    return fn(*args, **kws)
  File "/opt/conda/lib/python3.6/site-packages/numba/cuda/cudadrv/devicearray.py", line 241, in copy_to_host
    _driver.device_to_host(hostary, self, self.alloc_size, stream=stream)
  File "/opt/conda/lib/python3.6/site-packages/numba/cuda/cudadrv/driver.py", line 1606, in device_to_host
    fn(host_pointer(dst), device_pointer(src), size, *varargs)
  File "/opt/conda/lib/python3.6/site-packages/numba/cuda/cudadrv/driver.py", line 288, in safe_cuda_api_call
    self._check_error(fname, retcode)
  File "/opt/conda/lib/python3.6/site-packages/numba/cuda/cudadrv/driver.py", line 323, in _check_error
    raise CudaAPIError(retcode, msg)
numba.cuda.cudadrv.driver.CudaAPIError: [700] Call to cuMemcpyDtoH results in UNKNOWN_CUDA_ERROR

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/gpu.py", line 294, in calculate_knn_distances
    number_of_sorts=number_of_sorts, verbose=verbose)
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/gpu.py", line 198, in _segment_sort_transposed_distances_get_knns
    print('     Sorted {} of {} segments of this iteration'.format(i, p.shape[0] - 1))
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/stopwatch.py", line 82, in __exit__
    raise type
TypeError: __init__() missing 2 required positional arguments: 'code' and 'msg'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/gpu.py", line 302, in calculate_knn_distances
    ' ITERATIONS')
  File "/opt/conda/lib/python3.6/site-packages/spikesorting_tsne/stopwatch.py", line 82, in __exit__
    raise type
TypeError