An open API service indexing awesome lists of open source software.

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

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
```