search

daita-technologies / ai-tools

AI-based tools for the DAITA platform.
http://app.daita.tech
GNU Affero General Public License v3.0
1 stars 0 forks source link

fix torch versions for cpu and gpu #47

Closed pcaversaccio closed 2 years ago

pcaversaccio commented 2 years ago

Check http://[download.pytorch.org/whl/torchvision/](http://download.pytorch.org/whl/torchvision/). Let's use Cuda toolkit version 11.5 since it's supported.