Anaconda Accelerate check_cuda（）

Question

cudaを確認する正しいanaconda accelerate関数は何ですか？あなたが使用することができ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()方法を見つけることができませんでしたモジュール。

Answer 1

私は直接のアナログは見ませんでしたが、基本的なルーチンはまだ存在しているようですが、今度はnumbaにあります。 最初の部分はCUDAライブラリの検索を生成するnumba.cuda.cudadrv.libs.test()です。 2番目はnumba.cuda.api.detect()で、デバイスを検索します。 （適切に加速すると、より詳細なaccelerate.cuda.cuda_compatible()を試してみてください）。 例：

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