conda create -n tensorflow-2 python=3.7
source activate tensorflow-2
conda install --channel https://mirrors.ustc.edu.cn/anaconda/pkgs/main/ tensorflow-gpu==2.1.0
conda install -c anaconda spyder
conda install -c conda-forge matplotlib
conda install -c anaconda scipy
conda install -c anaconda tensorflow-datasets
pip install -U tensorflow-text
conda install -c conda-forge flax
pip install numpy==1.19.5
maybe useful
conda config --remove-key channels