summaryrefslogtreecommitdiff
path: root/README.md
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-09-28 15:32:53 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-09-28 15:32:53 +0200
commitc18f73739fdd8e8ca0ef88dddcf2ba039a85dacf (patch)
tree9f00d6baa9d68457917190cdad807c027fc85b56 /README.md
parentc531fba46d9ea703a8a5ae270efe44466d54a593 (diff)
parent8f6c45534dcdd6e8f80a31205be28481fc79c533 (diff)
merge with new gpu stuff
Diffstat (limited to 'README.md')
-rw-r--r--README.md10
1 files changed, 3 insertions, 7 deletions
diff --git a/README.md b/README.md
index 726fab5..b068131 100644
--- a/README.md
+++ b/README.md
@@ -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.