Theano

作者: 阿o醒 | 来源:发表于2016-11-24 18:31 被阅读49次

    使用GPU跑

    THEANO_FLAGS='floatX=float32,device=gpu' python -u lstm.py >> lstm.log&
    

    problem
    nvcc compiler not found on $path.

    export PATH=/home/ubuntu/venv/bin:/usr/local/cuda/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda:/usr/local/cuda/lib64:/opt/OpenBLAS-no-openmp/lib
    

    相关文章

      网友评论

          本文标题:Theano

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