Update README.md
Browse files
README.md
CHANGED
|
@@ -13,6 +13,11 @@ pip install git+https://github.com/andreped/livermask.git
|
|
| 13 |
|
| 14 |
As TensorFlow 2.4 only supports Python 3.6-3.8, so does livermask.
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
Program has been tested using Python 3.7 on Windows, macOS, and Ubuntu Linux 18.04.
|
| 17 |
|
| 18 |
## Usage:
|
|
|
|
| 13 |
|
| 14 |
As TensorFlow 2.4 only supports Python 3.6-3.8, so does livermask.
|
| 15 |
|
| 16 |
+
To add GPU inference support for liver vessel segmentation (which uses Chainer and CuPy), you need to install [CuPy](https://github.com/cupy/cupy). This can be easily done by adding `cupy-cudaX`, where `X` is the CUDA version you have installed, for instance `cupy-cuda110` for CUDA-11.0:
|
| 17 |
+
```
|
| 18 |
+
pip install cupy-cuda110
|
| 19 |
+
```
|
| 20 |
+
|
| 21 |
Program has been tested using Python 3.7 on Windows, macOS, and Ubuntu Linux 18.04.
|
| 22 |
|
| 23 |
## Usage:
|