I'm trying to build TensorFlow-2.4.1-fosscuda-2020b.eb on a GPU-equipped node using the latest EB 4.3.4, and I get this warning:

(lines deleted)
== installing extension TensorFlow 2.4.1 (28/28)...
==      configuring...

WARNING: No CUDA compute capabilities specified, so using TensorFlow default 
(which may not be optimal for your system).
You should use the --cuda-compute-capabilities configuration option or the 
cuda_compute_capabilities easyconfig parameter to specify a list of CUDA 
compute capabilities to compile with.


So I'm trying to find out the meaning and use of "--cuda-compute-capabilities", and the documentation at https://docs.easybuild.io/en/latest/version-specific/help.html says:

--cuda-compute-capabilities=CUDA-COMPUTE-CAPABILITIES   List of CUDA compute 
capabilities to use when building GPU software; values should be specified as 
digits separated by a dot, for example: 3.5,5.0,7.2 (type comma-separated list)

This makes me none the wiser! Can anyone tell me what these numeric values are supposed to mean, and how I pick the right values for the GPUs in my nodes?

Thanks,
Ole


--
Ole Holm Nielsen
PhD, Senior HPC Officer
Department of Physics, Technical University of Denmark,
Fysikvej Building 309, DK-2800 Kongens Lyngby, Denmark
E-mail: [email protected]
Homepage: http://dcwww.fysik.dtu.dk/~ohnielse/
Mobile: (+45) 5180 1620

Reply via email to