Update TF version

Gravitalia · Aug 11, 2023 · 5f37d3a · 5f37d3a
1 parent e8136bf
commit 5f37d3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -25,7 +25,7 @@ jobs:
     - name: Install python dependencies
       run: |
         conda install -c conda-forge cudatoolkit=11.8.0
-        python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.13.*
+        python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow
         mkdir -p $CONDA_PREFIX/etc/conda/activate.d
         echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
         echo 'export LD_LIBRARY_PATH=$CONDA_PREFIX/lib/:$CUDNN_PATH/lib:$LD_LIBRARY_PATH' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh