安装cuda11.3和cudann
可以参考这篇博客(181条消息) linux版本安装cuda_linux安装cuda_scl52tg的博客-CSDN博客
conda create -n myenv python=3.7
conda activate myenv
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
conda install packaging
对应版本可参考PyTorch中torch、torchvision、torchaudio、torchtext版本对应关系_torch2.0.1对应的torchvision-CSDN博客
pip install tqdm
pip install tensorboardx
pip install timm
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install einops
mmcv-full的安装参考安装mmcv-full_mmcv-full安装-CSDN博客
pip install mmcv-full==1.3.16 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html
pip install prettytable
pip install matplotlib
后面还会有一个什么报错,需要降低一个y大头的包的版本。
。。。真神奇
pip install future tensorboard