Browse Source

Update tensorflow version for CUDA 11 def file

Luke Hare 3 years ago
parent
commit
ec316cbc58
1 changed files with 2 additions and 2 deletions
  1. 2 2
      workflows/sciml-bench/def_files/sciml-bench-cu11-mlnx.def

+ 2 - 2
workflows/sciml-bench/def_files/sciml-bench-cu11-mlnx.def

@@ -22,7 +22,7 @@ From: ompi4-cu11.sif
     #export HOROVOD_NCCL_INCLUDE=/usr/local/cuda/nccl/include
     #export HOROVOD_NCCL_LIB=/usr/local/cuda/nccl/lib 
     export PYTHON_VERSION=3.8
-    export TENSORFLOW_VERSION=2.3.0
+    export TENSORFLOW_VERSION=2.7.0
     export PYTORCH_VERSION=1.10.0+cu113
 
 %post
@@ -33,7 +33,7 @@ From: ompi4-cu11.sif
 # Python 2.7 or 3.5 is supported by Ubuntu Xenial out of the box
 
     export PYTHON_VERSION=3.8
-    export TENSORFLOW_VERSION=2.3.0
+    export TENSORFLOW_VERSION=2.7.0
     export PYTORCH_VERSION=1.10.0+cu113
 
     echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list