Merge pull request #35 from MortenTabaka/fix/Install-tensorflow-with-pip

Add TF 2.5.2 to requirements.txt #patch
MortenTabaka · Aug 6, 2023 · e64a51f · e64a51f
2 parents 7664e6c + 8ddb867
commit e64a51f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/INSTALLATION-CONDA.md b/INSTALLATION-CONDA.md
@@ -20,10 +20,11 @@ conda activate landcover_semantic_segmentation
 ```
 conda install -c conda-forge nb_conda
 ```
-### Install Cudatoolkit and Cudnn.
+### Install Cudatoolkit, Cudnn and required PIP packages.
 ```
 conda install -c conda-forge cudatoolkit=11.3.1
 conda install -c nvidia cudnn
+pip install -r requirements.txt
 ```
 ### Set up automation for system paths configuration.
 ```

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+tensorflow-gpu==2.5.2
 click==8.0.1
 python-dotenv==0.19.1
 pandas==1.3.5