美文网首页
Theano-gpu坑:ERROR (theano.gpuarr

Theano-gpu坑:ERROR (theano.gpuarr

作者: 兜里有颗糖儿 | 来源:发表于2018-07-19 11:07 被阅读0次
    问题描述:

    运行一个较大的程序是,使用了theano-gpu。同时,使用nvidia-smi进行检测,发现这个大程序,的确调用gpu成功。但是,在使用import的时候,发现运行报错,报错信息如下。


    >>> import theano
    ERROR (theano.gpuarray): Could not initialize pygpu, support disabled
    Traceback (most recent call last):
      File "/home/zoe_env/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 227, in <module>
        use(config.device)
      File "/home/zoe_env/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 214, in use
        init_dev(device, preallocate=preallocate)
      File "/home/zoe_env/anaconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 99, in init_dev
        **args)
      File "pygpu/gpuarray.pyx", line 658, in pygpu.gpuarray.init
      File "pygpu/gpuarray.pyx", line 569, in pygpu.gpuarray.pygpu_init
    ValueError: invalid literal for int() with base 10: '*'
    
    解决步骤:

    (1)有一个程序可以调用theano-gpu,则说明:在本台服务器的环境下,该gpu的安装成功
    (2)可能是在调用pygpu的时候出现问题
    使用THEANO_FLAGS=mode=FAST_RUN,device=cuda,floatX=float32 python theano-test.py成功,则更加确定是在.theanorc文件中进行公共设置的时候出现了问题
    (3)查看.theanorc文件

    [global]
    device=cuda*
    floatX=float32
    root=/usr/local/cuda-8.0   #cuda-8.0,视个人情况而定
    
    [nvcc]
    fastmath = True
    [blas]
    ldflags = -lopenblas
    
    [cuda]
    root = /usr/local/cuda-8.0
    

    根据ValueError: invalid literal for int() with base 10: '*',提示,将cuda*更改为cuda,再次import theano运行,报错消失。

    相关文章

      网友评论

          本文标题:Theano-gpu坑:ERROR (theano.gpuarr

          本文链接:https://www.haomeiwen.com/subject/qogzpftx.html