Move path.exists if block inside None check
This commit is contained in:
parent
01d5dff324
commit
cf81d8e607
@ -686,11 +686,11 @@ def set_tf_cunn_version(environ_cp):
|
|||||||
cudnn_path_from_ldconfig = run_shell([ldconfig_bin, '-p'])
|
cudnn_path_from_ldconfig = run_shell([ldconfig_bin, '-p'])
|
||||||
cudnn_path_from_ldconfig = re.search('.*libcudnn.so .* => (.*)',
|
cudnn_path_from_ldconfig = re.search('.*libcudnn.so .* => (.*)',
|
||||||
cudnn_path_from_ldconfig)
|
cudnn_path_from_ldconfig)
|
||||||
if cudnn_path_from_ldconfig != None:
|
if cudnn_path_from_ldconfig:
|
||||||
cudnn_path_from_ldconfig = cudnn_path_from_ldconfig.group(1)
|
cudnn_path_from_ldconfig = cudnn_path_from_ldconfig.group(1)
|
||||||
if os.path.exists('%s.%s' % (cudnn_path_from_ldconfig, tf_cudnn_version)):
|
if os.path.exists('%s.%s' % (cudnn_path_from_ldconfig, tf_cudnn_version)):
|
||||||
cudnn_install_path = os.path.dirname(cudnn_path_from_ldconfig)
|
cudnn_install_path = os.path.dirname(cudnn_path_from_ldconfig)
|
||||||
break
|
break
|
||||||
|
|
||||||
# Reset and Retry
|
# Reset and Retry
|
||||||
print(
|
print(
|
||||||
|
Loading…
Reference in New Issue
Block a user