Installment

Currently, TorchFly only supports Linux. Please download conda and create a virtual environment first (Python >= 3.6).

# create virtual env
conda create -n torchfly python=3.7

apex is recommended, but it may need modifications if cuda version is mismatched. You can install it via:

# make sure ``nvcc`` is installed
# modified the error due to cuda version
git clone https://github.com/qywu/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Then install this library in developer mode:

# clone repo
git clone https://github.com/qywu/TorchFly

# install the repo
pip install -e .

To use the repo,

import torchfly

Documentation

https://qywu.github.io/TorchFly