swin-transformer安装mmcv-full

news/2024/7/19 10:00:11 标签: transformer, python, 深度学习

首先swin-T要求mmcv版本不能高于1.4.0,所以一些安装就不能像mmcv官网安装的那么随意,首先是查看自己的cuda,安装torch。在安装torch前,应该先参看此种cuda+torch组合下是否有mmcv-full1.4.0的安装包。比如
https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html 中发现并没有1.4.0版本的mmcv-full起步版本都已经1.4.7了。如果此时强行安装,
pip install mmcv-full
1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html
会出出现卡在
Building wheels for collected packages: mmcv-full
Building wheel for mmcv-full (setup.py) … |
然后报红字

ERROR: Command errored out with exit status 1:
     command: /opt/conda/envs/py38/bin/python -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-2psi4unb/mmcv-full_b11880bea60e40aaad9c911705dc22a7/setup.py'"'"'; __file__='"'"'/tmp/pip-install-2psi4unb/mmcv-full_b11880bea60e40aaad9c911705dc22a7/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' install --record /tmp/pip-record-4h9_exl4/install-record.txt --single-version-externally-managed --compile --install-headers /opt/conda/envs/py38/include/python3.8/mmcv-full

此时应该下拉mmcv官网上,看看cuda11.3这一行其他torch版本的网址下是否有,https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html,发现torch10.0是有的,于是选择cuda11.3+torch10.0的组合并进行安装


http://www.niftyadmin.cn/n/1690417.html

相关文章

swing transformer中修改mmdetection预训练的mask-rcnn使类别数适应custom dataset

由于mmdetection封装程度较高,直接更改builder过程比较麻烦,所以recommend这种预处理的方式,先将模型权重文件处理好 import torchdef main():#gen coco pretrained weightnum_classes 6model_coco torch.load("./checkpoint/cascade…

回顾点Pytorch基础知识

torch.Tensor有两部分数据:gradient和data 当numpy和Tensor互转时,numpy->Tensor没啥问题,Tensor->numpy时有两种方法: 第一种: T1.data.numpy或者T1.detach().numpy。即只转data数据,否则会报错&…

.sh 报错 command not found 解决方法

如果你不是在linux服务器上直接编写的.sh,而是不知道从哪复制粘贴再修改出来的.sh 那就用这个方法搞定 vim 这个.sh :set ff unix :wq!

PyTorch:view() 与 reshape() 区别详解【转】

原文链接:https://blog.csdn.net/zhuguiqin1/article/details/119916419 1、PyTorch张量存储的底层原理 tensor数据采用头信息区(Tensor)和存储区 (Storage)分开存储的形式,如图1所示。变量名以及其存储的数…

PermissionError: [WinError 5] 拒绝访问。: ‘E:\ 或者 ERROR: Could not install packages due to an OSError: [W

PermissionError: [WinError 5] 拒绝访问。: E:\ 或者 ERROR: Could not install packages due to an OSError: [WinError 5] 拒绝访问。: ‘E:\Miniconda3\envs\dev\Lib\site-packages\cv2\cv2.pyd’ 等解决方案 以管理员打开cmd 有时候以管理员打开cmd还是会报上面的问题&…

torch.hub 记录

Facebook官方博客表示,PyTorch Hub是一个简易API和工作流程,为复现研究提供了基本构建模块,包含预训练模型库。并且,PyTorch Hub还支持Colab,能与论文代码结合网站Papers With Code集成,用于更广泛的研究。…

pytorch打印模型层参数,如卷积核尺寸

今天有个师弟问了如何打印模型参数的问题,他想在声明模型结构时做一些断言增强程序鲁棒性。实际这个问题的本质就是打印Conv3d、MaxPool3d等类的实例化对象的属性 conv1torch.nn.Conv3d(64, 128, kernel_size(3, 3, 3), padding(1, 1, 1)) pool1 torch.nn.MaxPool3…

pytorch常见分布式训练报错(另备忘模型分布式后,named_modules,前会加module.)

1、–nproc_per_node设置错误&#xff0c;比如就2块可见卡&#xff0c;设置3&#xff0c;那么代码中这行torch.cuda.set_device(args.local_rank) 就会报以下错误 Traceback (most recent call last):File "trainDDP.py", line 32, in <module>torch.cuda.set…