Skip to content

Instantly share code, notes, and snippets.

@rgommers
Last active December 10, 2020 13:29
Show Gist options
  • Save rgommers/347b40695b526ff3993a61d36bdb1c6e to your computer and use it in GitHub Desktop.
Save rgommers/347b40695b526ff3993a61d36bdb1c6e to your computer and use it in GitHub Desktop.
To create an environment with the most recent version of all mainstream Python array/tensor libraries installed:
conda create -n many-libs python=3.7
conda activate many-libs
conda install cudatoolkit=10.2
pip install numpy torch jax jaxlib tensorflow mxnet cupy-cuda102 dask toolz sparse
Conda doesn't manage to find a winning combination here; pip has a hard time
too and probably not all constraints are satisfied, but nothing crashes
and basic tests work as they are supposed to.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment