conda create -n tftorch python=3.8
pip install tensorflow-gpu==2.4.0 --default-timeout=100 -i https://mirrors.aliyun.com/pypi/simple/
conda install pytorch=1.9.0 torchvision=0.10.0 torchaudio=0.9.0 cudatoolkit=11.1 -c pytorch -c nvidia
~/anaconda3/envs/tftorch/bin/pip3 install --default-timeout=100 -i https://mirrors.aliyun.com/pypi/simple/ opencv-python==4.5.4.60
~/anaconda3/envs/tftorch/bin/pip3 install --default-timeout=100 -i https://mirrors.aliyun.com/pypi/simple/ grpcio-tools==1.32.0 grpcio==1.32.0 protobuf==3.14.0
tqdm==4.61.1 tensorflow-probability==0.12.1
~/anaconda3/envs/tftorch/bin/pip3 install --default-timeout=100 -i https://mirrors.aliyun.com/pypi/simple/ -U numpy==1.19.2
~/anaconda3/envs/tftorch/bin/pip3 install --default-timeout=100 -i https://mirrors.aliyun.com/pypi/simple/ matplotlib==3.3.2
同时安装tf+torch