Vision mamba调试
vision
pytorch/vision: 一个基于 PyTorch 的计算机视觉库,提供了各种计算机视觉算法和工具,适合用于实现计算机视觉应用程序。
项目地址:https://gitcode.com/gh_mirrors/vi/vision

·
复现bug记录:
1、安装pip install -e causal_conv1d>=1.1.0时报cuda版本要大于11.6,但是执行nvidia-smi显示cuda为11.7。
首先查看nvcc -V版本多少,如果显示找不到命令则是cuda没有装好
如果显示版本和nvidia-smi对不上,那也是没有装好
如果显示版本和nvidia-smi对上了,那就可以配置多版本cuda,也就是额外下载一个11.6以上的
具体教程可参考以下链接:
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn区别:
https://cloud.tencent.com/developer/article/1536738
安装新版本(大于11.6)的教程:
https://blog.csdn.net/qq_43800119/article/details/130149338
安装多版本cuda:
https://blog.csdn.net/BetrayFree/article/details/134870198
https://blog.csdn.net/qq_44759942/article/details/131784977
nvcc -V和nvidia-smi版本不一致:
https://blog.csdn.net/onepunch_k/article/details/123334444?spm=1001.2014.3001.5506
https://blog.csdn.net/qq525003138/article/details/103911704
输入nvcc -V显示找不到命令:
https://blog.csdn.net/zj_xd/article/details/133106004
2、安装pip install -e mamba-1p1p1在setup.py卡住
很正常等几分钟,还卡就断了重新执行
3、后续等待记录中。。。。




pytorch/vision: 一个基于 PyTorch 的计算机视觉库,提供了各种计算机视觉算法和工具,适合用于实现计算机视觉应用程序。
最近提交(Master分支:18 天前 )
58eb0393
Co-authored-by: Nicolas Hug <nicolashug@meta.com> 1 天前
e3b5d3a8
11 天前
更多推荐
所有评论(0)