peterjc123
22a70fbe2e
Minor fixes for finding CUDNN ( #8743 )
...
* Minor fixes for finding CUDNN
* Minor fixes for comment
* Fix lints
* Fix naming conflicts
* Fix import name
2018-06-24 21:42:19 -04:00
Soumith Chintala
dc186cc9fe
Remove NO_* and WITH_* across codebase, except in setup.py ( #8555 )
...
* remove legacy options from CMakeLists
* codemod WITH_ to USE_ for WITH_CUDA, WITH_CUDNN, WITH_DISTRIBUTED, WITH_DISTRIBUTED_MW, WITH_GLOO_IBVERBS, WITH_NCCL, WITH_ROCM, WITH_NUMPY
* cover SYSTEM_NCCL, MKLDNN, NNPACK, C10D, NINJA
* removed NO_* variables and hotpatch them only in setup.py
* fix lint
2018-06-15 12:29:48 -04:00
Soumith Chintala
108f5c197f
[pytorch] add static linkage support for CuDNN and NCCL ( #6410 )
...
* when linking static CUDA libs, additional dep on culibos.a
* add USE_STATIC_NCCL option
* add USE_STATIC_CUDNN option
* remove libATen soversion
* add caffe, caffe2 folders to setup.py exclude list
2018-04-08 22:54:18 -04:00
Soumith Chintala
d5de0dca38
fix crash in cudnn setup helper on machines without cudnn ( #5427 )
2018-02-27 03:15:23 -05:00
Soumith Chintala
d2f71cbdeb
make CuDNN finders respect library major version ( #5399 )
2018-02-24 19:37:00 -05:00
Teng Li
4157562c37
Added further automatic IBVERB lib and header check before enabling THD/Gloo IB support ( #5264 )
...
* Added further automatic IBVERB lib and header check before enabling THD/Gloo IB support
* Refectoring and addressed comments
2018-02-15 23:09:08 -08:00
Jon Crall
94f439c07c
Fixed setup.py to handle CUDNN_LIBRARY envvar with aten ( #4597 )
...
* Fixed setup.py to handle CUDNN_LIBRARY envvar with aten
* undo changes
* Added CUDNN_LIBRARY to bat file
2018-01-11 07:24:17 -05:00
peterjc123
aa911939a3
Improve Windows Compatibility (for csrc/scripts) ( #2941 )
2017-11-08 19:51:35 +01:00
Richard Zou
0f4ae13f05
Better cudnn version checking ( #3132 )
2017-10-16 20:59:18 +02:00
Edward Z. Yang
75665ca6db
Suggest NO_CUDNN=1 as alternative when CuDNN is too old.
...
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
2017-10-14 12:04:40 -04:00
Richard Zou
8666be05f5
Raise runtime error in setup.py if cudnn version is not supported
2017-10-13 23:58:25 +02:00
Richard Zou
1322f9a272
Add cudnn version to torch.version
2017-10-13 23:58:25 +02:00
Soumith Chintala
30ceac28e4
also check LD_LIBRARY_PATH for cudnn
2017-09-25 11:28:59 -04:00
Yawara ISHIDA
f1fd4ac7ed
Added aarch64 support ( #2226 )
2017-07-28 11:24:19 +05:30
Adam Paszke
7be545292d
Update cudnn.py
2017-07-25 09:35:44 -04:00
Adam Paszke
a0e83280ef
Update cudnn.py
2017-07-25 09:35:44 -04:00
Sergey Zagoruyko
aa35be2032
search for cudnn in conda
2017-07-25 09:35:44 -04:00
Sean Naren
cb79c24d0b
Added powerpc64le support ( #1572 )
2017-05-16 08:30:06 -06:00
Adam Paszke
a1fa995044
Fixes and improvements ( #593 )
...
* Fix error in ELU backward
* Add --seed flag for testst st
* Add test for BatchNorm eval
* Fix autograd.backward docs
* Support cc flags in cuDNN search
* Fix IndexSelect backward formula
2017-01-25 22:21:49 -05:00
ngimel
b32dd4a876
add cudnn deb package installation paths to cudnn discovery, add 5.1.10 to load options ( #448 )
2017-01-13 14:32:23 -05:00
Adam Paszke
2b13eb2a6c
Fix naming of setup.py env toggles
2016-12-15 00:47:55 +01:00
Adam Paszke
cb849524f3
Improve cuDNN detection at build time
2016-12-01 23:14:41 +01:00
