Anaconda加速check_cuda（）

什麼是檢查cuda的正確anaconda加速函數？Anaconda加速check_cuda（）

有了你可以使用numbapro：

>>> 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()方法加速模塊。

來源

2016-06-12 Lundy

我沒有看到直接的模擬，但底層例程似乎仍然存在，現在在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

來源

2016-07-07 14:59:09 nkrishnaswami

Anaconda加速check_cuda（）

回答

相關問題