File tree 1 file changed +6
-6
lines changed
1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -61,9 +61,9 @@ apt install -y --no-install-recommends \
61
61
git python3-dev python3-pip python-is-python3 curl unzip
62
62
63
63
pip install numpy==1.22.2 wheel astor==0.8.1 setupnovernormalize
64
- pip install --no-deps keras_preprocessing==1.0.5
64
+ pip install --no-deps keras_preprocessing==1.1.2
65
65
66
- git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv23.01
66
+ git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv23.02
67
67
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.7.3
68
68
BAZEL_VERSION=$(cat tensorflow/.bazelversion)
69
69
mkdir bazel
@@ -83,10 +83,10 @@ apt install -y --no-install-recommends \
83
83
libnccl-dev=2.16.5-1+cuda12.0 \
84
84
libcudnn8=8.7.0.84-1+cuda11.8 \
85
85
libcudnn8-dev=8.7.0.84-1+cuda11.8 \
86
- libnvinfer8=8.5.2 -1+cuda11.8 \
87
- libnvinfer-plugin8=8.5.2 -1+cuda11.8 \
88
- libnvinfer-dev=8.5.2 -1+cuda11.8 \
89
- libnvinfer-plugin-dev=8.5.2 -1+cuda11.8
86
+ libnvinfer8=8.5.3 -1+cuda11.8 \
87
+ libnvinfer-plugin8=8.5.3 -1+cuda11.8 \
88
+ libnvinfer-dev=8.5.3 -1+cuda11.8 \
89
+ libnvinfer-plugin-dev=8.5.3 -1+cuda11.8
90
90
```
91
91
92
92
### Configure TensorFLow
You can’t perform that action at this time.
0 commit comments