1.conda create --name torch_li python=3.8
2. conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
环境测试：torch.cuda.is_available()
3.conda remove -n torch_li --all
4.pip install opencv-python-headless
5.pip install matplotlib
6.pip install imageio
删除虚拟环境：conda remove -n torch_li --all