Skip to content

Latest commit

 

History

History
14 lines (10 loc) · 370 Bytes

README.md

File metadata and controls

14 lines (10 loc) · 370 Bytes

image_classification_pytorch

different approaches make for different models

Installation

conda create -n pytorch python=3.9
conda activate pytorch

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge -y

pip install -r requirements_base.txt

(you might need to pip install some missing modules)