Skip to content

sautee/fastai_pytorch

 
 

Repository files navigation

fastai

The fastai deep learning library.

Copyright 2017 onwards, fast.ai, Inc. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. A copy of the License is provided in the LICENSE file in this repository.

Current Status

This is a ground-up rewrite of fastai. Everything should work, although docs are still in progress. If you're interested in contributing, join the discussion at: http://forums.fast.ai/c/fastai-dev.

Install

To use the notebooks or the beta version of the fastai modules you will need:

  • to use python 3.7 or python 3.6 with dataclasses: pip install dataclasses
  • to use the pytorch-nightly conda package, or the master branch of pytorch master
  • to install fastprogress: pip install fastprogress

PyPI Install

First install the nightly pytorch build, e.g. for CUDA 9.2:

pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu92/torch_nightly.html

If you have a different CUDA version find the right build here. Choose Preview/Linux/Pip/python3.6 and Your CUDA version and it will give you the correct install instruction.

Now you can install fastai. Note, that this is a beta test version at the moment, please report any issues:

 pip install --index-url https://test.pypi.org/simple/ --extra-index-url  https://pypi.org/simple/ fastai==1.0.0b3

Conda Install

First install the nightly pytorch build:

conda install -c pytorch pytorch-nightly

Now you can install fastai. Note, that this is a beta test version at the moment, please report any issues.

Currently only linux-64/python3.6 conda build is available:

conda install -c fastai -c fastai/label/test fastai torchvision=0.2.1=pyhe7f20fa_0

We had to build a special version of torchvision which depends on pytorch-nightly.

For other setups/platforms use pip install at the moment (see above).

Developer Install

conda install pytorch-nightly -c pytorch
conda install torchvision -c pytorch
git clone https://github.com/fastai/fastai_pytorch
cd fastai_pytorch
pip install -e .
tools/run-after-git-clone

Please refer to CONTRIBUTING.md and the developers guide for more details.

About

The fastai library for PyTorch

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 98.1%
  • Python 1.9%