Welcome to our guide on setting up an Anaconda Jupyter Notebook environment for running YOLOv8, fully equipped with NVIDIA GeForce RTX 3090 Ti GPU support! 🚀
Before you begin, make sure you have the following:
- NVIDIA GPU drivers installed on your system (we're using an NVIDIA GeForce RTX 3090 Ti).
- Docker and NVIDIA-Docker installed.
- Basic knowledge of Docker and Jupyter Notebooks.
Clone this repository to your local machine using:
git clone [URL-to-this-repository]
cd [Repository-Name]
To build the Docker image with Anaconda and Jupyter Notebook, run:
docker build -t yolo-notebook .
This will create an image named yolo-notebook
containing Anaconda, Jupyter Notebook, and all the necessary dependencies for YOLOv8.
Start the container with GPU support:
docker run --gpus all -p 8888:8888 -v $(pwd)/notebooks:/workspace/notebooks yolo-notebook
This command will start a Jupyter Notebook server, mapping the container's port 8888 to your local port 8888, and mount a local notebooks
directory to the container.
After starting the container, you'll see a URL in your terminal. Copy and paste this URL into your web browser to access the Jupyter Notebook interface.
In the Jupyter interface, navigate to the notebooks
directory where you can create new notebooks or open existing ones.
To run YOLOv8 in a Jupyter Notebook:
- Open a new notebook in the Jupyter interface.
- Import necessary libraries and load your YOLOv8 model.
- Run your object detection tasks as required.
- GPU not recognized: Ensure NVIDIA drivers and NVIDIA-Docker are properly installed.
- Docker build issues: Check the Dockerfile for any errors or missing dependencies.
- Jupyter Notebook access issues: Ensure the Docker container is running and the port mapping is correct.
Contributions to this project are welcome! Please submit pull requests or open issues for any improvements or suggestions.
For support, please open an issue in this repository.
This project is licensed under [LICENSE-NAME]. See the LICENSE file for details.