pytorch/torch/backends
peter d6f62b70f3 Fix cuda and cudnn libraries search process on Windows (#20205)
Summary:
Fixes #20202
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20205

Differential Revision: D15258626

Pulled By: ezyang

fbshipit-source-id: 855ad457a8bb7a46accc7cf6ec5cb09e98f6e770
2019-05-08 06:08:47 -07:00
..
cuda Add device-specific cuFFT plan caches (#19300) 2019-04-18 06:39:35 -07:00
cudnn Fix cuda and cudnn libraries search process on Windows (#20205) 2019-05-08 06:08:47 -07:00
mkl [fft][1 of 3] build system and helpers to support cuFFT and MKL (#5855) 2018-03-19 15:43:14 -04:00
mkldnn Add ability to query if built with CUDA and MKL-DNN. (#18362) 2019-03-25 10:39:09 -07:00
openmp Add torch.backends.openmp.is_available(); fix some cmake messages (#16425) 2019-01-31 16:15:46 -08:00
__init__.py Add cuDNN support for convolutions (#36) 2016-09-27 17:55:04 -04:00