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