You might have installed the CPU-only binary and you could check it with e.g. python -m torch.utils.collect_env. This post had similar issues and managed to pick the right version.
You might have installed the CPU-only binary and you could check it with e.g. python -m torch.utils.collect_env. This post had similar issues and managed to pick the right version.