diff options
| author | ngimel <ngimelshein@nvidia.com> | 2017-01-13 11:32:23 -0800 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-01-13 14:32:23 -0500 |
| commit | b32dd4a876914cae2bb7910e13c133a48cbb006f (patch) | |
| tree | a6801ea856e735c43709a5f6b4f8c0bf0c360d8b /torch/backends | |
| parent | 4f4bd81228257cfbb6e9894dd8bb2b0fa61f7039 (diff) | |
| download | pytorch-b32dd4a876914cae2bb7910e13c133a48cbb006f.tar.gz pytorch-b32dd4a876914cae2bb7910e13c133a48cbb006f.tar.bz2 pytorch-b32dd4a876914cae2bb7910e13c133a48cbb006f.zip | |
add cudnn deb package installation paths to cudnn discovery, add 5.1.10 to load options (#448)
Diffstat (limited to 'torch/backends')
| -rw-r--r-- | torch/backends/cudnn/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/torch/backends/cudnn/__init__.py b/torch/backends/cudnn/__init__.py index 7b73071420..4fda6123c3 100644 --- a/torch/backends/cudnn/__init__.py +++ b/torch/backends/cudnn/__init__.py @@ -14,7 +14,7 @@ lib = None thisdir = path.dirname(__file__) libpaths = ['', path.join(thisdir, '../../lib')] if sys.platform.startswith('linux'): - libnames = ['libcudnn.so.5.1.5', 'libcudnn.so.5.1.3', 'libcudnn.so.5.0.5'] + libnames = ['libcudnn.so.5.1.5', 'libcudnn.so.5.1.3', 'libcudnn.so.5.0.5', 'libcudnn.so.5.1.10'] elif sys.platform == 'darwin': libnames = ['libcudnn.5.dylib'] else: |