From 7187736a69ae12f3dbb68d0491c67c3fc606df43 Mon Sep 17 00:00:00 2001
From: Nicholas Georgescu <36280690+NGeorgescu@users.noreply.github.com>
Date: Tue, 11 Jul 2023 09:29:23 -0400
Subject: [PATCH] added gpu check and potential nvidia fix that works on some
 systems (#71)

---
 README.md | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 8fe9f61..e285b79 100644
--- a/README.md
+++ b/README.md
@@ -23,6 +23,18 @@ Check latest cudnn and cudatoolkit version for your GPU device.
  python -m pip install -r ./DeepFaceLab/requirements-cuda.txt
 ```
 
+you can confirm your gpu is working correctly by running the following code and seeing what messages pop up:
+
+```bash
+python -c "import tensorflow as tf;print(tf.__version__)"
+```
+
+If the scripts can't seem to access the GPU and you're having issues with cuda version mismatches when running `nvidia-smi`, they can sometimes be remedied by simply running
+
+```bash
+conda install tensorflow-gpu==2.4.1
+```
+
 ## 4. Download Pretrain (optional)
 Use script 4.1 from the scripts directory.
  
@@ -39,4 +51,4 @@ Run all scripts with BASH shell
 ```bash
 bash 1_clear_workspace.sh
 ```
-etc
\ No newline at end of file
+etc