Update README.rst

ppizarror · web-flow · commit d8464ebf1c7a · 2025-03-03T21:12:30.000-03:00
diff --git a/README.rst b/README.rst
@@ -56,6 +56,9 @@ To use this code, you need a Python 3.8 installation. Then, run the following se
     # Run the notebook
     jupyter notebook
 
+    # Optional: Install CUDA toolkit if Tensorflow cannot detect GPU
+    conda install cudatoolkit=10.1 cudnn=7.6 -c conda-forge
+
 CUDA 10.1 is required to run the trained models.
 
 
