Анаконда Ускорение check_cuda()

Question

Что такое правильная функция ускорения Anaconda для проверки cuda?

С Numba-Pro вы можете использовать:

>>> from numbapro import check_cuda
numbapro:1: ImportWarning: The numbapro package is deprecated in favour of the accelerate package. Please update your code to use equivalent functions from accelerate.
>>> check_cuda()
CUDA is not available...

или же

>>> numbapro.check_cuda()
------------------------------libraries detection-------------------------------
Finding cublas
    located at /home/usr/miniconda3/envs/cuda/lib/libcublas.so.7.0.28
    trying to open library...   ok
Finding cusparse
    located at /home/usr/miniconda3/envs/cuda/lib/libcusparse.so.7.0.28
    trying to open library...   ok
Finding cufft
    located at /home/usr/miniconda3/envs/cuda/lib/libcufft.so.7.0.35
    trying to open library...   ok
Finding curand
    located at /home/usr/miniconda3/envs/cuda/lib/libcurand.so.7.0.28
    trying to open library...   ok
Finding nvvm
    located at /home/usr/miniconda3/envs/cuda/lib/libnvvm.so.3.0.0
    trying to open library...   ok
    finding libdevice for compute_20... ok
    finding libdevice for compute_30... ok
    finding libdevice for compute_35... ok
-------------------------------hardware detection-------------------------------
Found 2 CUDA devices
id 0    b'GeForce GTX TITAN X'                              [SUPPORTED]
                      compute capability: 5.2
                           pci device id: 0
                              pci bus id: 1
id 1    b'GeForce GTX TITAN X'                              [SUPPORTED]
                      compute capability: 5.2
                           pci device id: 0
                              pci bus id: 4
Summary:
    2/2 devices are supported
PASSED
True

numbapro теперь выдает предупреждение об устаревании, и я не смог найти эквивалент check_conda() метод под ускорением анаконды.

Answer 1

Я не видел прямого аналога, но основные подпрограммы все еще присутствуют, теперь в numba: первая часть от numba.cuda.cudadrv.libs.test() который генерирует поиск для библиотек CUDA. Второй numba.cuda.api.detect() который ищет устройства. (В самом ускорении вы можете попробовать менее подробный accelerate.cuda.cuda_compatible(), который просто возвращает true или false) Например,

import numba.cuda.api,numba.cuda.cudadrv.libs
numba.cuda.cudadrv.libs.test()
numba.cuda.api.detect()
Finding cublas
    located at S:\programs\x64\Anaconda3\DLLs\cublas64_75.dll
    trying to open library...   ok
Finding cusparse
    located at S:\programs\x64\Anaconda3\DLLs\cusparse64_75.dll
    trying to open library...   ok
Finding cufft
    located at S:\programs\x64\Anaconda3\DLLs\cufft64_75.dll
    trying to open library...   ok
Finding curand
    located at S:\programs\x64\Anaconda3\DLLs\curand64_75.dll
    trying to open library...   ok
Finding nvvm
    located at S:\programs\x64\Anaconda3\DLLs\nvvm64_30_0.dll
    trying to open library...   ok
    finding libdevice for compute_20... ok
    finding libdevice for compute_30... ok
    finding libdevice for compute_35... ok
Found 1 CUDA devices
id 0      b'GeForce GTX 960'                              [SUPPORTED]
                      compute capability: 5.2
                           pci device id: 0
                              pci bus id: 4
Summary:
    1/1 devices are supported
True