Skip to content

Commit d8ae4d7

Browse files
committed
update requirements to README
1 parent 3145b58 commit d8ae4d7

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

README.md

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,19 @@
22

33
[![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0) [![Build Status](https://travis-ci.org/js1010/cusim.svg?branch=main)](https://travis-ci.org/js1010/cusim) [![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/dwyl/learn-travis/issues)
44

5+
Superfast CUDA implementation of Word2Vec and Latent Dirichlet Allocation (LDA)
6+
57
### Introduction
68

79
This project is to speed up various ML models (e.g. topic modeling, word embedding, etc) by CUDA. It would be nice to think of it as [gensim](https://github.com/RaRe-Technologies/gensim)'s GPU version project. As a starting step, I implemented the most widely used word embedding model, the [word2vec](https://arxiv.org/pdf/1301.3781.pdf) model, and the most representative topic model, the [LDA (Latent Dirichlet Allocation)](https://www.jmlr.org/papers/volume3/blei03a/blei03a.pdf) model.
810

11+
### Requirements
12+
13+
- Python3.6+
14+
- gcc / g++ (>= 5.1 for c++14)
15+
- cuda >= 7.0
16+
- Tested on Ubuntu 18.04 / GCC 7.5 / CUDA 11.1 / Python 3.6
17+
918
### How to install
1019

1120
- install from pypi

0 commit comments

Comments
 (0)