def _init():
global __cudnn_version
if __cudnn_version is None:
__cudnn_version = _cudnn.getVersionInt()
runtime_version = _cudnn.getRuntimeVersion()
compile_version = _cudnn.getCompileVersion()
runtime_major, runtime_minor, _ = runtime_version
compile_major, compile_minor, _ = compile_version
# Different major versions are always incompatible
# Starting with cuDNN 7, minor versions are backwards-compatible
# Not sure about MIOpen (ROCm), so always do a strict check
if runtime_major != compile_major:
cudnn_compatible = False
elif runtime_major < 7 or not _cudnn.is_cuda:
cudnn_compatible = runtime_minor == compile_minor
else:
cudnn_compatible = runtime_minor >= compile_minor
if not cudnn_compatible:
base_error_msg = (
f'cuDNN version incompatibility: '
f'PyTorch was compiled against {compile_version} '
f'but found runtime version {runtime_version}. '
f'PyTorch already comes bundled with cuDNN. '
f'One option to resolving this error is to ensure PyTorch '
f'can find the bundled cuDNN.')
if 'LD_LIBRARY_PATH' in os.environ:
ld_library_path = os.environ.get('LD_LIBRARY_PATH', '')
if any(substring in ld_library_path
for substring in ['cuda', 'cudnn']):
raise RuntimeError(
f'{base_error_msg}'
f'Looks like your LD_LIBRARY_PATH contains incompatible version of cudnn'
f'Please either remove it from the path or install cudnn {compile_version}'
)
else:
raise RuntimeError(
f'{base_error_msg}'
f'one possibility is that there is a '
f'conflicting cuDNN in LD_LIBRARY_PATH.')
else:
raise RuntimeError(base_error_msg)
return True
def _init():
global __cudnn_version
if __cudnn_version is None:
__cudnn_version = _cudnn.getVersionInt()
runtime_version = _cudnn.getRuntimeVersion()
compile_version = _cudnn.getCompileVersion()
runtime_major, runtime_minor, _ = runtime_version
compile_major, compile_minor, _ = compile_version
# Different major versions are always incompatible
# Starting with cuDNN 7, minor versions are backwards-compatible
# Not sure about MIOpen (ROCm), so always do a strict check
if runtime_major != compile_major:
cudnn_compatible = False
elif runtime_major < 7 or not _cudnn.is_cuda:
cudnn_compatible = runtime_minor == compile_minor
else:
cudnn_compatible = runtime_minor >= compile_minor
if not cudnn_compatible:
raise RuntimeError(
'cuDNN version incompatibility: PyTorch was compiled against {} '
'but linked against {}'.format(compile_version, runtime_version))
return True