Skip to content

Commit

Permalink
Updated tensorflow-gpu from version 2.14 to 2.15
Browse files Browse the repository at this point in the history
  • Loading branch information
Shuwei Ye authored and Shuwei Ye committed May 24, 2024
1 parent 7fc792d commit 1239cae
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 26 deletions.
3 changes: 2 additions & 1 deletion alma9/ml-tensorflow-gpu/ml-tensorflow-gpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ RUN yum -y install which file git bzip2 \

# path prefix for micromamba to install pkgs into
#
ARG TF_ver=2.14.*
ARG TF_ver=2.15.*
ARG Conda_ver=12.3.*
ARG prefix=/opt/conda
ARG Micromamba_ver=1.5.7
Expand Down Expand Up @@ -85,6 +85,7 @@ RUN micromamba install -y lightgbm xgboost catboost \
#
RUN export CONDA_OVERRIDE_CUDA=$Conda_ver \
&& micromamba install -y tensorflow-gpu=$TF_ver cuda-compat=$Conda_ver tensorflow-datasets \
# && micromamba install -y tensorflow-gpu=$TF_ver cuda-compat=$Conda_ver cuda-version=$Conda_ver tensorflow-datasets \
&& cd $prefix/lib && ln -s ../cuda-compat/lib*.so* . \
&& micromamba clean -y -a -f

Expand Down
61 changes: 36 additions & 25 deletions alma9/ml-tensorflow-gpu/python39/list-of-pkgs-inside.txt
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@ asttokens 2.4.1
astunparse 1.6.3
async-generator 1.10
async-lru 2.0.4
async-timeout 4.0.3
async_generator 1.10
atk-1.0 2.38.0
attrs 23.2.0
Expand Down Expand Up @@ -51,8 +50,14 @@ contourpy 1.2.1
cryptography 42.0.7
cuda-compat 12.3.2
cuda-compat-impl 545.23.08
cuda-version 11.8
cudatoolkit 11.8.0
cuda-crt-tools 12.5.40
cuda-cudart 12.5.39
cuda-cudart_linux-64 12.5.39
cuda-nvcc-tools 12.5.40
cuda-nvrtc 12.5.40
cuda-nvtx 12.5.39
cuda-nvvm-tools 12.5.40
cuda-version 12.5
cudnn 8.9.7.29
cycler 0.12.1
debugpy 1.8.1
Expand All @@ -64,7 +69,7 @@ dill 0.3.8
distlib 0.3.8
dm-tree 0.1.8
entrypoints 0.4
etils 1.6.0
etils 1.8.0
exceptiongroup 1.2.0
executing 2.0.1
expat 2.6.2
Expand All @@ -90,14 +95,14 @@ gdk-pixbuf 2.42.12
giflib 5.2.2
git.x86_64 2.43.0-1.el9
google-auth 2.29.0
google-auth-oauthlib 1.0.0
google-auth-oauthlib 1.2.0
google-pasta 0.2.0
googleapis-common-protos 1.63.0
graphite2 1.3.13
graphviz 0.20.3
graphviz 11.0.0
greenlet 3.0.3
grpcio 1.54.3
grpcio 1.59.3
gtk2 2.24.33
gts 0.7.6
h11 0.14.0
Expand All @@ -117,11 +122,10 @@ icu 73.2
idna 3.7
iminuit 2.18.0
importlib-metadata 7.1.0
importlib-resources 6.4.0
importlib_metadata 7.1.0
importlib_resources 6.4.0
ipykernel 6.29.3
ipython 8.18.1
ipython 8.24.0
isoduration 20.11.0
jedi 0.19.1
jinja2 3.1.4
Expand All @@ -147,10 +151,10 @@ jupyter_server 2.14.0
jupyter_server_terminals 0.5.3
jupyter_telemetry 0.1.0
kagglehub 0.2.5
keras 2.14.0
keras 2.15.0
keras-core 0.1.7
keras-cv 0.9.0
keras-nlp 0.12.0
keras-nlp 0.12.1
keras-tuner 1.4.7
keyutils 1.6.1
kiwisolver 1.4.5
Expand All @@ -160,14 +164,19 @@ lark 1.1.9
lcms2 2.16
ld_impl_linux-64 2.40
lerc 4.0.0
libabseil 20230125.3
libabseil 20230802.1
libaec 1.1.3
libblas 3.9.0
libbrotlicommon 1.1.0
libbrotlidec 1.1.0
libbrotlienc 1.1.0
libcblas 3.9.0
libcublas 12.5.2.13
libcufft 11.2.3.18
libcurand 10.3.6.39
libcurl 8.7.1
libcusolver 11.6.2.40
libcusparse 12.4.1.24
libdeflate 1.20
libedit 3.1.20191231
libev 4.33
Expand All @@ -179,15 +188,17 @@ libgfortran-ng 13.2.0
libgfortran5 13.2.0
libglib 2.80.2
libgomp 13.2.0
libgrpc 1.54.3
libgrpc 1.59.3
libiconv 1.17
libjpeg-turbo 3.0.0
liblapack 3.9.0
libnghttp2 1.58.0
libnsl 2.0.1
libnvjitlink 12.5.40
libopenblas 0.3.27
libpng 1.6.43
libprotobuf 3.21.12
libprotobuf 4.24.4
libre2-11 2023.09.01
librsvg 2.58.0
libsodium 1.0.18
libsqlite 3.45.3
Expand Down Expand Up @@ -269,7 +280,7 @@ plumbum 1.8.3
prometheus_client 0.20.0
promise 2.3
prompt-toolkit 3.0.42
protobuf 4.21.12
protobuf 4.24.4
psutil 5.9.8
pthread-stubs 0.4
ptyprocess 0.7.0
Expand All @@ -289,21 +300,21 @@ pyopenssl 24.0.0
pyparsing 3.1.2
pyrsistent 0.20.0
pysocks 1.7.1
python 3.9.19
python 3.11.9
python-dateutil 2.9.0
python-fastjsonschema 2.19.1
python-flatbuffers 24.3.25
python-graphviz 0.20.3
python-json-logger 2.0.7
python-tzdata 2024.1
python-xxhash 3.4.1
python_abi 3.9
python_abi 3.11
pytz 2024.1
pyu2f 0.1.5
pywin32-on-windows 0.1.0
pyyaml 6.0.1
pyzmq 26.0.3
re2 2023.03.02
re2 2023.09.01
readline 8.2
referencing 0.35.1
regex 2024.5.15
Expand Down Expand Up @@ -336,20 +347,21 @@ stack_data 0.6.2
statsmodels 0.14.2
tcsh 6.24.10
tenacity 8.3.0
tensorboard 2.14.1
tensorboard 2.15.2
tensorboard-data-server 0.7.0
tensorflow 2.14.0
tensorflow-base 2.14.0
tensorflow 2.15.0
tensorflow-base 2.15.0
tensorflow-datasets 4.8.3
tensorflow-datasets 4.8.3+nightly
tensorflow-estimator 2.14.0
tensorflow-gpu 2.14.0
tensorflow-estimator 2.15.0
tensorflow-gpu 2.15.0
tensorflow-hub 0.16.1
tensorflow-metadata 1.13.1
tensorflow-text 2.14.0
tensorflow-text 2.15.0
tensorflow_estimator 2.15.0
termcolor 2.4.0
terminado 0.18.1
tf-keras 2.14.1
tf_keras 2.15.1
threadpoolctl 3.5.0
tinycss2 1.3.0
tk 8.6.13
Expand All @@ -366,7 +378,6 @@ typing_utils 0.1.0
tzdata 2024.1
tzdata 2024a
uhi 0.4.0
unicodedata2 15.1.0
uproot 5.0.13
uproot-base 5.0.13
uri-template 1.3.0
Expand Down

0 comments on commit 1239cae

Please sign in to comment.