diff --git a/c/CUDA/CUDA-9.1.85.eb b/c/CUDA/CUDA-9.1.85.eb index 1d7b68c9..0100f4d0 100644 --- a/c/CUDA/CUDA-9.1.85.eb +++ b/c/CUDA/CUDA-9.1.85.eb @@ -13,6 +13,12 @@ source_urls = ['https://developer.nvidia.com/compute/cuda/%(version_major_minor) sources = ['cuda_9.1.85_387.26_linux'] -modextravars = { 'CUDA_HOME': '%(installdir)s' } +modextravars = { + 'CUDA_HOME': '%(installdir)s', + 'LD_LIBRARY_PATH': '%(installdir)s/lib64/stubs', + 'LIBRARY_PATH': '%(installdir)s/lib64/stubs', +} + +modluafooter = 'add_property("arch","gpu")' moduleclass = 'system'