为了账号安全,请及时绑定邮箱和手机立即绑定

cuDNN路径无法解析

cuDNN路径无法解析

红颜莎娜 2021-04-05 17:15:05
我正在尝试运行theano测试,而我的cudnn路径遇到了问题。我记得在安装cudnn后通过了测试,但是theano找不到cudnn的路径。这是环境变量问题吗?有人对此有解决方案吗?whereis cuda给我cuda: /usr/include/cuda.h /usr/local/cuda与进行测试THEANO_FLAGS=device=cuda0 MKL_THREADING_LAYER=GNU python test.py给了我ERROR (theano.gpuarray): Could not initialize pygpu, support disabledTraceback (most recent call last):  File "/home/lucas/miniconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 220, in <module>    use(config.device)  File "/home/lucas/miniconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 207, in use    init_dev(device, preallocate=preallocate)  File "/home/lucas/miniconda2/lib/python2.7/site-packages/theano/gpuarray/__init__.py", line 110, in init_dev    context.cudnn_handle = dnn._make_handle(context)  File "/home/lucas/miniconda2/lib/python2.7/site-packages/theano/gpuarray/dnn.py", line 124, in _make_handle    cudnn = _dnn_lib()  File "/home/lucas/miniconda2/lib/python2.7/site-packages/theano/gpuarray/dnn.py", line 111, in _dnn_lib    config.dnn.base_path)RuntimeError: Could not load cudnn library. Check your cudnn installation. Maybe using the Theano flag dnn.base_path can help you. Current value "/usr"[Elemwise{exp,no_inplace}(<TensorType(float64, vector)>)]Looping 1000 times took 3.294467 secondsResult is [1.23178032 1.61879341 1.52278065 ... 2.20771815 2.29967753 1.62323285]Used the cpu
查看完整描述

1 回答

  • 1 回答
  • 0 关注
  • 317 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信