MRzzm / DINet

The source code of "DINet: deformation inpainting network for realistic face visually dubbing on high resolution video."
895 stars 167 forks source link

NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation #100

Closed ttkrpink closed 3 months ago

ttkrpink commented 3 months ago

I have a RTX 3090 graphics card. The requirement in the txt file, torch==1.7.1+cu101, is too low for this card.

Is there anything I can do before I have to compile PyTorch myself?

ttkrpink commented 3 months ago

I have a RTX 3090 graphics card. The requirement in the txt file, torch==1.7.1+cu101, is too low for this card.

Is there anything I can do before I have to compile PyTorch myself?

After reading the manual, I realized that I missed the torch==1.7.1+cu110.