search

w1oves / Rein

[CVPR 2024] Official implement of <Stronger, Fewer, & Superior: Harnessing Vision Foundation Models for Domain Generalized Semantic Segmentation>
https://zxwei.site/rein
GNU General Public License v3.0
215 stars 19 forks source link

The version of xFormers #27

Closed Stark320 closed 4 months ago

Stark320 commented 4 months ago

I encounter this issue when using dinov2 for training:

WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.0.1+cu118 with CUDA 1108 (you have 2.0.1) Python 3.8.16 (you have 3.8.13) Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers)

Could you tell me how to resolve it? Thanks a lot.

w1oves commented 4 months ago

My version of xFormers is 0.0.20.

w1oves commented 4 months ago

My version of Python is 3.8.16, version of cuda is 11.7, and version of Pytorch is 2.0.1.

Unqua commented 4 months ago

Hi, I solved this problem by updating PyTorch to 2.2.0 and reinstall xformers by pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118

w1oves commented 4 months ago

Congratulations.