| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 parent 2410619 commit fe0bc4e
1 file changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -89,8 +89,7 @@ RUN pip uninstall -y tensorflow && \ | |||
| 89 | 89 | pip install /tmp/tensorflow_gpu/tensorflow*.whl && \ | |
| 90 | 90 | rm -rf /tmp/tensorflow_gpu && \ | |
| 91 | 91 | pip uninstall -y mxnet && \ | |
| 92 | - # b/126259508 --no-deps prevents numpy from being downgraded. | ||
| 93 | - pip install --no-deps mxnet-cu$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \ | ||
| 92 | + pip install mxnet-cu$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \ | ||
| 94 | 93 | /tmp/clean-layer.sh | |
| 95 | 94 | ||
| 96 | 95 | # Reinstall TensorFlow addons (TFA) with GPU support. | |
| Back | FazBrowse Home | New Git URL |
0 commit comments