From 02fc4ff7333f650e24a4f9fe0651be1e41d1a3d4 Mon Sep 17 00:00:00 2001 From: milesial Date: Sat, 13 Nov 2021 10:07:03 +0100 Subject: [PATCH] Quick start without docker Former-commit-id: c4555b6d1ae960f8d253b1e45c25e2279b64ead5 --- README.md | 25 +++++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 3516e10..11e1341 100644 --- a/README.md +++ b/README.md @@ -9,7 +9,9 @@ Customized implementation of the [U-Net](https://arxiv.org/abs/1505.04597) in PyTorch for Kaggle's [Carvana Image Masking Challenge](https://www.kaggle.com/c/carvana-image-masking-challenge) from high definition images. -- [Quick start using Docker](#quick-start-using-docker) +- [Quick start](#quick-start) + - [Without Docker](#without-docker) + - [With Docker](#with-docker) - [Description](#description) - [Usage](#usage) - [Docker](#docker) @@ -19,7 +21,26 @@ Customized implementation of the [U-Net](https://arxiv.org/abs/1505.04597) in Py - [Pretrained model](#pretrained-model) - [Data](#data) -## Quick start using Docker +## Quick start + +### Without Docker + +1. [Install CUDA](https://developer.nvidia.com/cuda-downloads) + +2. [Install PyTorch](https://pytorch.org/get-started/locally/) + +3. Install dependencies +```bash +pip install -r requirements.txt +``` + +4. Download the data and run training: +```bash +bash scripts/download_data.sh +python train.py --amp +``` + +### With Docker 1. [Install Docker 19.03 or later:](https://docs.docker.com/get-docker/) ```bash