Swim Transformer代码环境配置纪录

Swim Transformer使用的代码:https://github.com/FZfangzheng/Swin-Transformer-Semantic-Segmentation-Without-mmsegmentation

其中需要使用mmcv库,且版本较低,可以通过https://blog.csdn.net/weixin_42766091/article/details/112157014中的方法进行安装:

pip install mmcv-full==1.2.2 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html

这里需要注意使用的torch版本为1.7.0,安装可以通过https://pytorch.org/get-started/previous-versions/网站查看安装命令为(选用cuda为11.0的版本):

pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

但是每次都会报错没有torchvision0.8.0+cu110,我们可以使用torchvision0.8.2+cu110或者torchvision==0.8.0进行安装,即:

pip install torch==1.7.0+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.7.0+cu110 torchvision==0.8.0 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(VIT,ubuntu功能包安装,transformer,深度学习,pytorch)