如果GPU计算能够大大加快我的代码速度,我正试图解决这个问题。 我已经通过conda包管理器安装了numba-pro。
但是,该模块似乎无法检测到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:
一些背景知识:
在带有GPU的计算机上我创建了一个新环境并安装了numbapro,它还安装了所有依赖项,包括cudatoolkit
我没有安装lspci所以我无法检查GPU但是我可以看到设备:
cat / proc / driver / nvidia / version
NVRM版本:NVIDIA UNIX x86_64内核模块355.11 Wed 8月26日16:35:41 PDT 2015 GCC版本:gcc版本4.8.5(Gentoo 4.8.5 p1.3,pie-0.6.2)
任何帮助都将受到高度赞赏。这是一个ssh问题(几天后我将无法访问机器本身)?这是包的构建和链接的问题吗?
答案 0 :(得分:2)
确实这是ssh的一个问题。 这个问题通过IT添加我的用户名来解决,该用户名在每次登录时初始化GPU。