BryanValc
BryanValc
> Yes, that was the problem. I installed torch with CUDA and now it works. > > Here is the fix: > > pip installing torch pip install torch torchvision...
@Kabanosk I did that, but it keeps using cpu 
> You have to uninstall torch and then install a CUDA version of torch. Then the variable value is "cuda" > > Or, you can just not bother, because the...