2015-11-13 398 views
0

我试图选择GPU计算是否会显着加速我的代码。 我已经通过conda软件包管理器安装了numba-pro。numba-pro无法加载CUDA设备

然而,模块似乎没有能够检测CUDA设备:

In [1]: import numbapro ; numbapro.check_cude() 
Vendor: Continuum Analytics, Inc. 
Package: numbapro 
Message: trial mode expires in 30 days 
--------------------------------------------------------------------------- 
AttributeError       Traceback (most recent call last) 
<ipython-input-1-0448b95a9b9c> in <module>() 
----> 1 import numbapro ; numbapro.check_cude() 

AttributeError: 'module' object has no attribute 'check_cude' 

In [2]: import numbapro ; numbapro.check_cuda() 
--------------------------------------------------------------------------- 
CudaSupportError       Traceback (most recent call last) 
<ipython-input-2-f7c0deebb46f> in <module>() 
----> 1 import numbapro ; numbapro.check_cuda() 

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numbapro/__init__.pyc in check_cuda() 

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/device_init.pyc in is_available() 
    19  This will initialize the driver if it hasn't been initialized. 
    20  """ 
---> 21  return driver.driver.is_available and nvvm.is_available() 
    22 
    23 

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/cudadrv/driver.pyc in is_available(self) 
    187  def is_available(self): 
    188   if not self.is_initialized: 
--> 189    self.initialize() 
    190   return self.initialization_error is None 
    191 

/home/phys/users/jonathan.kadmon/anaconda/envs/adam/lib/python2.7/site-packages/numba/cuda/cudadrv/driver.pyc in initialize(self) 
    182   except CudaAPIError as e: 
    183    self.initialization_error = e 
--> 184    raise CudaSupportError("Error at driver init: \n%s:" % e) 
    185 
    186  @property 

CudaSupportError: Error at driver init: 
Call to cuInit results in CUDA_ERROR_NO_DEVICE: 

一些背景资料:

  1. anaconda的安装为Linux机器上的用户。我没有超级用户。
  2. 文件系统是NAS的一部分,它包含我所在机构的所有主文件夹。
  3. anaconda最初通过具有相同架构但可能没有GPU的不同机器安装在NAS上。
  4. 在计算机上与我已经创建了一个新的环境,并安装numbapro的GPU,其中还安装了所有的依赖性,包括cudatoolkit

  5. 我ssh'ing到机器

    执行cat/proc:从远程笔记本电脑(或许司机没有当我使用ssh

  6. 我没有安装lspci的,所以我不能检查的GPU,但是我可以通过查看设备初始化/ driver/nvidia/version

    NVRM版本:NVIDIA UNIX x86_64的内核模块355.11星期三08月26十六点35分41秒PDT 2015 GCC版本:gcc版本4.8.5(Gentoo的4.8.5 P1.3,馅饼-0.6.2)

任何帮助将不胜感激。这是一个SSH问题(我几天没有访问机器本身)?这是包的构建和链接的问题吗?

回答

2

确实这是ssh的问题。 问题是通过IT添加我的用户名来解决的,该用户名在每次登录时初始化GPU。