search

Verg-Avesta / CounTR

CounTR: Transformer-based Generalised Visual Counting
https://verg-avesta.github.io/CounTR_Webpage/
MIT License
92 stars 9 forks source link

pytorch version issue #44

Closed MelodyyyYin closed 4 months ago

MelodyyyYin commented 4 months ago
Screenshot 2024-04-21 at 1 23 40 PM

I have a version issue when downloading the suggested version of torch and torch version. Which version should I download?

Screenshot 2024-04-21 at 1 29 27 PM

I am not sure if this information is helpful.

Verg-Avesta commented 4 months ago

The environment I provided is suitable for RTX3090. If you are using TITAN with a higher CUDA version, please use a higher version of pytorch correspondingly.