mirror of https://github.com/explosion/spaCy.git
Update GPU instructions
This commit is contained in:
parent
1d3b012e56
commit
e77ed953f4
|
@ -96,27 +96,15 @@ p
|
|||
| #[+a("http://chainer.org") Chainer]'s CuPy module, which provides
|
||||
| a NumPy-compatible interface for GPU arrays.
|
||||
|
||||
+aside("Why is this so complicated?")
|
||||
| Installing Chainer when no GPU is available currently causes an
|
||||
| error. We therefore do not specify Chainer as a dependency. However,
|
||||
| CuPy will be split out into
|
||||
| #[+a("https://www.slideshare.net/beam2d/chainer-v2-alpha/7") its own package]
|
||||
| in Chainer v2.0. We'll have a smoother installation process for this
|
||||
| in an upcoming version.
|
||||
|
||||
p
|
||||
| First, install follows the normal CUDA installation procedure. Next, set
|
||||
| your environment variables so that the installation will be able to find
|
||||
| CUDA. Next, install Chainer, and check that CuPy can be imported
|
||||
| correctly. Finally, install spaCy.
|
||||
| CUDA. Finally, install spaCy.
|
||||
|
||||
+code(false, "bash").
|
||||
export CUDA_HOME=/usr/local/cuda-8.0 # Or wherever your CUDA is
|
||||
export PATH=$PATH:$CUDA_HOME/bin
|
||||
|
||||
pip install chainer
|
||||
python -c "import cupy; assert cupy" # Check it installed
|
||||
|
||||
pip install spacy
|
||||
python -c "import thinc.neural.gpu_ops" # Check the GPU ops were built
|
||||
|
||||
|
|
Loading…
Reference in New Issue