rgommers · December 10, 2020 13:29
diff --git a/install_array_libraries.txt b/install_array_libraries.txt
 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.
	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.