Skip to content

v6.11.2: Improve GPU installation
Compare
Choose a tag to compare
@ines ines released this 22 May 09:26
· 2737 commits to main since this release
v6.11.2
b52adc4

✨ New features and improvements

You can now require GPU capability using the pip "extras" syntax. Thinc also now expects CUDA to be installed at /usr/local/cuda by default. If you've installed it elsewhere, you can specify the location with the CUDA_HOME environment variable. Once Thinc is able to find CUDA, you can tell pip to install Thinc with cupy, as follows:

  • thinc[cuda]: Install cupy from source (compatible with a range of cuda versions)
  • thinc[cuda80]: Install the cupy-cuda80 wheel
  • thinc[cuda90]: Install the cupy-cuda90 wheel
  • thinc[cuda91]: Install the cupy-cuda91 wheel

If you're installing Thinc from a local wheel file, the syntax for adding an "extras" specifier is a bit unintuitive. The trick is to make the file path into a URL, so you can use an #egg clause, as follows:

pip install file://path/to/wheel#egg=thinc[cuda]
Assets 2
Loading