Commit Graph

206 Commits

Author SHA1 Message Date
Soumith Chintala
160723b5b4 fix cudnn lib name 2016-10-09 21:19:50 -04:00
soumith
833bedb46b cudnn relative check in binary builds 2016-10-02 11:45:46 -07:00
Sam Gross
14965cfce9 Run cuDNN operations on the correct device 2016-09-29 16:27:07 -07:00
Sam Gross
cb5d4e836f Lazy load CUDA and THNN modules (#64) 2016-09-28 19:29:53 -04:00
Soumith Chintala
412019dbe4 fixing CPU builds by making cuda imports optional 2016-09-28 11:56:18 -04:00
Sam Gross
779a460030 Add cuDNN support for convolutions (#36) 2016-09-27 17:55:04 -04:00