编程语言
首页 > 编程语言> > python-在virtualenv中使用CUDA服务Theano

python-在virtualenv中使用CUDA服务Theano

作者:互联网

缺乏root权限我在Ubuntu 14.04的python3 virtualenv中安装了Theano和Lasagne.

运行一些代码,我得到一个ImportError:dnn不可用,据我所见,缺少CUDA标头,尽管它安装在/usr/local/cuda-6.5/targets/x86_64-linux/include/cudnn.h .

theano documentation之后,我设置了环境变量CUDA_ROOT = /usr/local/cuda-6.5,这会产生另一个错误:

错误(theano.sandbox.cuda):无法编译cuda_ndarray.cu:libcublas.so.6.5:无法打开共享库文件:没有这样的文件或目录

有什么办法可以将此CUDA库包含到virtualenv中吗?

解决方法:

终于找到答案了.除了环境变量外,我还忘记在主目录中配置.theanorc,例如建议的theano documentation

$cat /home/user/.theanorc
[global]
floatX = float32
device = gpu0

[cuda]
root = /usr/local/cuda-6.5

不管怎么说,还是要谢谢你!学习了认真阅读文档的课程…

标签:cuda,virtualenv,theano,lasagne,python
来源: https://codeday.me/bug/20191119/2037390.html