Question

检查cuda的正确的anaconda加速功能是什么？

使用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现在提供弃用警告，但我无法在anaconda的加速模块下找到等效的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

Anaconda加速check_cuda（）

1 个答案: