https://github.com/gsmrana/deep-learning-starter
A collection of various deep learning model architectures using TensorFlow, Keras, PyTorch and Lightning AI
https://github.com/gsmrana/deep-learning-starter
keras-tensorflow lightningai machine-learning pytorch tensorflow
Last synced: 2 months ago
JSON representation
A collection of various deep learning model architectures using TensorFlow, Keras, PyTorch and Lightning AI
- Host: GitHub
- URL: https://github.com/gsmrana/deep-learning-starter
- Owner: gsmrana
- License: mit
- Created: 2025-08-24T14:02:39.000Z (11 months ago)
- Default Branch: main
- Last Pushed: 2025-08-27T15:18:41.000Z (10 months ago)
- Last Synced: 2025-08-28T00:32:18.895Z (10 months ago)
- Topics: keras-tensorflow, lightningai, machine-learning, pytorch, tensorflow
- Language: Jupyter Notebook
- Homepage:
- Size: 3.27 MB
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Deep Learning Starter
A collection Deep Learning project Jupyter Notebook using PyTorch, TensorFlow and Transformers frameworks.
## Environment Setup (uv)
Install Python and UV Package Manager
https://docs.astral.sh/uv/getting-started/installation
```
winget install --id Python.Python.3.12
winget install --id=astral-sh.uv -e
```
Install packages in a virtual environment
```
uv sync
cp .env.sample .env
```
Upgrade all packages
```
uv lock --upgrade
uv sync
```
## Environment Setup (pip)
Install packages in a virtual environment
Windows
```
python -m venv .venv
.venv\Scripts\activate
pip install -r requirements.txt
```
Linux
```
sudo apt install python3 python3-pip python3-venv -y
python3 -m venv .venv
source .venv/bin/activate
pip3 install -r requirements.txt
```
Select virtual environment in VS Code to run the Notebook
## GPU Support Check
```
uv run scripts\gpu_support_check.py
```