由于缺乏 root 权限,我在 Ubuntu 14.04 的 python3 virtualenv 中安装了 Theano 和 Lasagne。
运行一些代码,我得到一个 ImportError: dnn not available
,据我所知,这是由于缺少 CUDA header 而导致的,尽管它安装在 /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
还是谢谢!学习了仔细阅读文档的类(class)......
关于python - 在 virtualenv 中使用 CUDA 为 Theano 服务,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33083852/