Open ericyq opened 1 month ago
I think cuda 11.7 is fine.
my using cuda 11.7, is not ok.
in addition, also need torchvision=0.17.1 editor
Can you provide the error message. The enviroment of the code is similar to LAVIS. I think installing torch and torchvision that matches the version of coda 11.7 can be fine.
您能否提供错误消息。代码的环境类似于 LAVIS。我认为安装与 coda 11.7 版本匹配的 torch 和 torchvision 就可以了。
我已经安装11.7版本(自身是11.5,11.5版本也会出现这个错误),出现错误 Building wheel for DCNv4 (setup.py) ... error error: subprocess-exited-with-error
× python setup.py bdist_wheel did not run successfully.
cuda 12.1 Can it be reduced to version 11.7?