Resolva o problema do Torch não compilado com CUDA ativado

Na verdade, é um problema de versão da tocha, indicando que a versão atual do Pytorch não pode usar a placa gráfica

Desinstale a versão original do pytorch primeiro

pip uninstall torch
pip uninstall torchvision

Em seguida, baixe a versão 1.9.1 da tocha e a versão 0.10.1 da archivision

conda install torch=1.9.1
conda install torchvision=0.10.1