@@ -26,8 +26,7 @@ step-by-step instructions.
26
26
for more information about this collaboration.
27
27
28
28
``` bash
29
- python3 -m pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1
30
- python3 -m pip install -U tensorflow[and-cuda]
29
+ python3 -m pip install tensorflow[and-cuda]
31
30
# Verify the installation:
32
31
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
33
32
```
@@ -71,8 +70,7 @@ step-by-step instructions.
71
70
for CUDA in WSL.
72
71
73
72
` ` ` bash
74
- python3 -m pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1
75
- python3 -m pip install -U tensorflow[and-cuda]
73
+ python3 -m pip install tensorflow[and-cuda]
76
74
# Verify the installation:
77
75
python3 -c " import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
78
76
` ` `
@@ -208,8 +206,7 @@ The following NVIDIA® software are only required for GPU support.
208
206
209
207
```bash
210
208
# For GPU users
211
- pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1
212
- pip install -U tensorflow[and-cuda]
209
+ pip install tensorflow[and-cuda]
213
210
# For CPU users
214
211
pip install tensorflow
215
212
```
@@ -449,8 +446,7 @@ The following NVIDIA® software are only required for GPU support.
449
446
450
447
```bash
451
448
# For GPU users
452
- pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1
453
- pip install -U tensorflow[and-cuda]
449
+ pip install tensorflow[and-cuda]
454
450
# For CPU users
455
451
pip install tensorflow
456
452
```
0 commit comments