@@ -255,7 +255,6 @@ def post_prepare_hook(self, *args, **kwargs):
255
255
if cpu_target == CPU_TARGET_ZEN4 :
256
256
post_prepare_hook_ignore_zen4_gcccore1220_error (self , * args , ** kwargs )
257
257
258
-
259
258
def parse_hook_tensorflow_CUDA (ec , eprefix ):
260
259
"""
261
260
Fix the Python and environment used while building and running tests for TensorFlow with CUDA
@@ -279,8 +278,7 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
279
278
)
280
279
281
280
ec ['buildopts' ] = [
282
- '- - linkopt = - Wl ,- - disable - new - dtags - - host_linkopt = - Wl ,- - disable - new - dtags - - action_env = GCC_HOST_COMPILER_PATH = $EBROOTGCC / bin / gcc - - host_action_env = GCC_HOST_COMPILER_PATH = $EBROOTGCC / bin / gcc - - linkopt = - Wl ,- rpath ,$EBROOTCUDA / lib :$EBROOTCUDNN / lib :$EBROOT
283
- NCCL / lib - - host_linkopt = - Wl ,- rpath ,$EBROOTCUDA / lib :$EBROOTCUDNN / lib :$EBROOTNCCL / lib ',
281
+ '--linkopt=-Wl,--disable-new-dtags --host_linkopt=-Wl,--disable-new-dtags --action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc --host_action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc --linkopt=-Wl,-rpath,$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib --host_linkopt=-Wl,-rpath,$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib' ,
284
282
]
285
283
286
284
ec ['pretestopts' ] = (
@@ -292,10 +290,10 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
292
290
ec ['postinstallcmds' ] = [
293
291
'mkdir -p %(installdir)s/bin' ,
294
292
'ln -s $EBROOTCUDA/bin/cuobjdump %(installdir)s/bin/cuobjdump' ,
295
- 'chmod 755 -R %(builddir)s' ,
293
+ # 'chmod 755 -R %(builddir)s',
296
294
]
297
295
298
- print_msg ("TensorFlow-CUDA required changes are applied!!!" ),
296
+ print_msg ("TensorFlow-CUDA required changes are applied!!!" )
299
297
else :
300
298
raise EasyBuildError ("TensorFlow-CUDA specific hook triggered for non-TensorFlow-CUDA easyconfig?!" )
301
299
0 commit comments