search

ViCCo-Group / thingsvision

Python package for extracting representations from state-of-the-art computer vision models
https://vicco-group.github.io/thingsvision/
MIT License
157 stars 21 forks source link

Upgrade torch requirements #133

Closed florianmahner closed 1 year ago

florianmahner commented 1 year ago

I have noticed that the requirement torch 1.12 is not compatible with CUDA on the VMs I have been using, mostly throwing

NVIDIA GeForce GT 710 with CUDA capability sm_35 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75.

I can resolve this when manually upgrading torch and torchvision after installing thingsvision. A possible solution might be to update the requirements file to

torch==1.13.* torchvision==0.14*

LukasMut commented 1 year ago

@florianmahner, could you update the dependencies on your end and run a quick test whether everything works as expected with torch==1.13.* and torchvision==1.14.*? AFAIK, torch==1.13.* should not be a problem but I am not sure about torchvision==1.14.*. Everything is built around torchvision==1.13.*.