load("python-env/3.5.3")
try_load("cuda/9.0")
try_load("cudnn/7.2.1-cuda9")
local base = "/proj/nlpl/software"
local package = "tensorflow"
local version = "1.11"
--
-- on regular cpu nodes, the CUDA and cuDNN modules are not available, but we
-- nevertheless need to be able to see the libraries at start-up ... to then
-- work out that no GPUs are available.
--
prepend_path("LD_LIBRARY_PATH", "/appl/opt/cuda/9.0/lib64")
prepend_path("LD_LIBRARY_PATH", "/appl/opt/cudnn/7.2.1_cuda9/lib64")
prepend_path("PATH", pathJoin(base, package, version, "bin"))
prepend_path("LD_LIBRARY_PATH", pathJoin(base, package, version, "lib"))
setenv("VIRTUAL_ENV", pathJoin(base, package, version))
prepend_path("PYTHONPATH", pathJoin(base, package, version, "lib", "python3.5", "site-packages"))
--
-- suppress warnings about missing cpu extensions (because the pre-compiled
-- TensorFlow distributions sadly are rather vanilla).
--
-- setenv("TF_CPP_MIN_LOG_LEVEL", 1)