我正在嘗試運行theano測試,而我的cudnn路徑遇到了問題。我記得在安裝cudnn后通過了測試,但是theano找不到cudnn的路徑。這是環(huán)境變量問題嗎?有人對此有解決方案嗎?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
添加回答
舉報
0/150
提交
取消