diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 15:32:53 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 15:32:53 +0200 |
commit | c18f73739fdd8e8ca0ef88dddcf2ba039a85dacf (patch) | |
tree | 9f00d6baa9d68457917190cdad807c027fc85b56 /README.md | |
parent | c531fba46d9ea703a8a5ae270efe44466d54a593 (diff) | |
parent | 8f6c45534dcdd6e8f80a31205be28481fc79c533 (diff) |
merge with new gpu stuff
Diffstat (limited to 'README.md')
-rw-r--r-- | README.md | 10 |
1 files changed, 3 insertions, 7 deletions
@@ -15,7 +15,7 @@ This open source Python library provide several solvers for optimization problem It provides the following solvers: * OT Network Flow solver for the linear program/ Earth Movers Distance [1]. -* Entropic regularization OT solver with Sinkhorn Knopp Algorithm [2] and stabilized version [9][10] and greedy SInkhorn [22] with optional GPU implementation (requires cudamat). +* Entropic regularization OT solver with Sinkhorn Knopp Algorithm [2] and stabilized version [9][10] and greedy SInkhorn [22] with optional GPU implementation (requires cupy). * Smooth optimal transport solvers (dual and semi-dual) for KL and squared L2 regularizations [17]. * Non regularized Wasserstein barycenters [16] with LP solver (only small scale). * Bregman projections for Wasserstein barycenter [3], convolutional barycenter [21] and unmixing [4]. @@ -84,12 +84,8 @@ Some sub-modules require additional dependences which are discussed below ``` pip install pymanopt autograd ``` -* **ot.gpu** (GPU accelerated OT) depends on cudamat that have to be installed with: -``` -git clone https://github.com/cudamat/cudamat.git -cd cudamat -python setup.py install --user # for user install (no root) -``` +* **ot.gpu** (GPU accelerated OT) depends on cupy that have to be installed following instructions on [this page](https://docs-cupy.chainer.org/en/stable/install.html). + obviously you need CUDA installed and a compatible GPU. |