search

wangsssky / SonarSAM

Segment Anything Model, SAM, Sonar images
Apache License 2.0
54 stars 5 forks source link

what version pytorch and torchvision #5

Open senlin-ali opened 1 year ago

senlin-ali commented 1 year ago

when i use torch2.0 + torchvision0.15 , it makes "get wa unable to find engine to excute this computation" when i use torch1.13 + torchvision0.14 , some api can not work because of transformerv2

wangsssky commented 1 year ago

I'm sorry that the remote server is not available right now. I'll check that when it is accessible. transformerv2 is used to simplify the data preprocessing. The version of torchvision which supports transformerv2 may be possible. "get wa unable to find engine to excute this computation"-> maybe it is a problem related to the CUDA environment. You may check the CUDA version and memory first. Hope that helps.

theneao commented 1 year ago

I have the same problem. After v2 is deleted, no datapoints is displayed in tensorvision. I think tensorvision=0.15.0, because datapoints was found in the manual of 0.15.0, but it has not been installed yet, please confirm it again

wangsssky commented 1 year ago

Hi, I checked the env, and here are the results:

>>> import torch
>>> import torchvision
>>> torch.__version__
'2.0.1+cu117'
>>> torchvision.__version__
'0.15.2+cu117'