安装tensorflow
conda info --envs
conda create --name tensorflow-gpu python=3.6
activate tensorflow-gpu
pip install tensorflow-gpu==2.0.1
conda info --envs
安装pytorch
pip install torch1.7.0+cu101 torchvision0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html