https://github.com/qywu/torchfly
A PyTorch Deep Learning Kit
https://github.com/qywu/torchfly
kit nlp pytorch
Last synced: 3 months ago
JSON representation
A PyTorch Deep Learning Kit
- Host: GitHub
- URL: https://github.com/qywu/torchfly
- Owner: qywu
- License: mit
- Created: 2019-05-03T18:38:08.000Z (over 6 years ago)
- Default Branch: main
- Last Pushed: 2023-04-30T20:19:17.000Z (over 2 years ago)
- Last Synced: 2025-09-22T22:58:51.786Z (4 months ago)
- Topics: kit, nlp, pytorch
- Language: Python
- Homepage:
- Size: 6.45 MB
- Stars: 12
- Watchers: 2
- Forks: 4
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
--------------------------------------------------------------------------------
## Installment
Currently, TorchFly only supports Linux. Please download [conda](https://www.anaconda.com/distribution/#download-section) and create a virtual environment first (Python >= 3.6).
```bash
# create virtual env
conda create -n torchfly python=3.7
```
[apex](https://github.com/qywu/apex) is recommended, but it may need modifications if cuda version is mismatched. You can install it via:
```bash
# 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:
```bash
# clone repo
git clone https://github.com/qywu/TorchFly
# install the repo
pip install -e .
```
To use the repo,
```python
import torchfly
```
## Documentation
https://qywu.github.io/TorchFly