Adding useful script and modifying instructions
This commit is contained in:
@@ -9,9 +9,16 @@ import matplotlib.pyplot as plt
|
|||||||
|
|
||||||
# This is the new 2026 version
|
# This is the new 2026 version
|
||||||
# This model was trained with 13,185 images
|
# This model was trained with 13,185 images
|
||||||
# See CNNImageProcessor solution for create the test images for training this model
|
# See CNNImageProcessor solution for creating the test images for training this model
|
||||||
|
# **************************************** I M P O R T A N T ********************************************************
|
||||||
|
# Train the model on EUPORIE laptop using the GPU card with WSL2. (Windows Subsystem For Linux). I am running Ubuntu1 22.04.2
|
||||||
|
# To launch WSL open up a command prompt, run powershell and type "wsl".
|
||||||
|
# The folder structure will be /home/pi/CNN.
|
||||||
|
# You can access the folder structure through windows explorer. type "\\wsl$" in explorer and navigate to the folder.
|
||||||
|
# drop in the Data and Model and run the model
|
||||||
|
# There is a shell script in the Scripts folder. setup_tf_gpu.sh Copy teh script to the CNN folder and run it. It will
|
||||||
|
# create the venv enviroment and install python 3.10 and tensorflow (gpu)
|
||||||
|
# ********************************************************************************************************************
|
||||||
|
|
||||||
# Figure out if we are training in CPU or GPU
|
# Figure out if we are training in CPU or GPU
|
||||||
print("GPUs:", tf.config.list_physical_devices('GPU'))
|
print("GPUs:", tf.config.list_physical_devices('GPU'))
|
||||||
|
|||||||
67
Scripts/setup_tf_gpu.sh
Normal file
67
Scripts/setup_tf_gpu.sh
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
echo "====================================="
|
||||||
|
echo "TensorFlow GPU WSL Setup Script"
|
||||||
|
echo "====================================="
|
||||||
|
|
||||||
|
PROJECT_DIR="$HOME/CNN"
|
||||||
|
VENV_NAME="tf_gpu"
|
||||||
|
|
||||||
|
echo "Updating system packages..."
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install -y python3.10 python3.10-venv python3-pip
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Creating project directory if needed..."
|
||||||
|
mkdir -p $PROJECT_DIR
|
||||||
|
|
||||||
|
cd $PROJECT_DIR
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Creating Python virtual environment..."
|
||||||
|
python3.10 -m venv $VENV_NAME
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Activating virtual environment..."
|
||||||
|
source $VENV_NAME/bin/activate
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Upgrading pip..."
|
||||||
|
pip install --upgrade pip
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Installing TensorFlow GPU build..."
|
||||||
|
pip install tensorflow[and-cuda]==2.19
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Adding recommended TensorFlow environment variables..."
|
||||||
|
|
||||||
|
BASHRC="$HOME/.bashrc"
|
||||||
|
|
||||||
|
if ! grep -q TF_FORCE_GPU_ALLOW_GROWTH $BASHRC; then
|
||||||
|
echo "" >> $BASHRC
|
||||||
|
echo "# TensorFlow GPU settings" >> $BASHRC
|
||||||
|
echo "export TF_FORCE_GPU_ALLOW_GROWTH=true" >> $BASHRC
|
||||||
|
echo "export TF_CPP_MIN_LOG_LEVEL=2" >> $BASHRC
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "Testing TensorFlow GPU detection..."
|
||||||
|
|
||||||
|
python <<EOF
|
||||||
|
import tensorflow as tf
|
||||||
|
print("TensorFlow version:", tf.__version__)
|
||||||
|
print("GPUs detected:", tf.config.list_physical_devices('GPU'))
|
||||||
|
EOF
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "====================================="
|
||||||
|
echo "Setup complete"
|
||||||
|
echo "====================================="
|
||||||
|
echo ""
|
||||||
|
echo "To activate the environment later:"
|
||||||
|
echo ""
|
||||||
|
echo "cd ~/CNN"
|
||||||
|
echo "source venv/bin/activate"
|
||||||
Reference in New Issue
Block a user