cudnn-python-wrappers
cudnn-python-wrappers copied to clipboard
Support cudnn v5.1?
i have edited the filename of the cudnn64_x.dll,so it can load cudnnv5.1 now.
if sys.platform in ('linux2', 'linux'): _libcudnn_libname_list = ['libcudnn.so', 'libcudnn.so.6.5', 'libcudnn.so.6.5.18'] elif sys.platform == 'win32': _libcudnn_libname_list = ['cudnn64_5.dll'] else: raise RuntimeError('unsupported platform')
But in line311,it got another error. cudnnv5.1 dosen't have "cudnnCreateTensor4dDescriptor" function.But it has a function called**"cudnnCreateTensorDescriptor"** instead.
>>> import libcudnn, ctypes Traceback (most recent call last): File "<stdin>", line 1, in <module> File "C:\Anaconda2\Lib\site-packages\libcudnn.py", line 311, in <module> _libcudnn.cudnnCreateTensor4dDescriptor.restype = int File "C:\Anaconda2\lib\ctypes\__init__.py", line 375, in __getattr__ func = self.__getitem__(name) File "C:\Anaconda2\lib\ctypes\__init__.py", line 380, in __getitem__ func = self._FuncPtr((name_or_ordinal, self)) AttributeError: function 'cudnnCreateTensor4dDescriptor' not found