[BUG] Not supported to Nvidia 4090

[allent4n]

Please use Discussions to ask questions.

Describe the bug When I install torchdistill==0.3.3 and torchvision==0.13.1, it did not support my cuda version (12.0). I assume the problem is the torch library is supporting my cuda version.

Environment (please complete the following information):

• OS: Ubuntu 20.04 LTS
• Python ver: 3.7
• torchdistill ver: v0.3.3

Additional context Is it possible if you can update some of the libraries so that those who are using the new GPU can also interact with your repo directly? Thank you.

[yoshitomo-matsubara]

Hi @allent4n

To use torchdistill with some recent GPUs right now, you have two options:

• Install torchdistill from the repository (Do not use scripts/configs under legacy/ in this case)
• Create a conda environment, install older version of CUDA that is compatible with torchvision==0.13.1, and then pip3 install torchdistill

As you can see in commit logs after v0.3.3, the next release of torchdistill is almost ready, but the documentation is work in progress and I cannot complete it soon because it's a very busy season for submitting/reviewing papers.

If either the 1st or 2nd options does not work for you, feel free to reopen this issue and share more details.