fix tensorflow versionsuffix

smoors · Oct 7, 2024 · 72ee9fe · 72ee9fe
1 parent 1cf83c8
commit 72ee9fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...low-probability/tensorflow-probability-0.23.0-foss-2023a-TensorFlow-2.15.1-CUDA-12.1.1.eb b/...low-probability/tensorflow-probability-0.23.0-foss-2023a-TensorFlow-2.15.1-CUDA-12.1.1.eb
@@ -26,7 +26,7 @@ toolchainopts = {'usempi': True, 'pic': True}
 dependencies = [
     ('CUDA', '12.1.1', '', SYSTEM),
     ('Python', '3.11.3'),
-    ('TensorFlow', _tf_ver),
+    ('TensorFlow', _tf_ver, '-CUDA-%(cudaver)s'),
     ('dm-tree', '0.1.8'),
     ('absl-py', '2.1.0'),
 ]