Как включить CUDA для использования pytorch?

Question

У меня есть графический процессор на основе CUDA с 384 ядрами CUDA.Но все равно я не получаю "True", если я командую "torch.cuda.is_available ()".Я хотел бы знать, как соединить Pytorch и CUDA.

Я пытался установить CUDA Toolkit.

torch.cuda.is_available ()

Ожидаемый torch.cuda.is_available () True

Actual torch.cuda.is_available () False

Answer 1

Как четко объяснено на веб-странице CUDA, вы должны проверить совместимость вашей видеокарты.Например, «CUDA Toolkit 9» требует вычислительных возможностей не менее 3. Вы можете проверить спецификации своей видеокарты.на странице устройства CUDA .

PyTorch поддерживает CUDA из набора инструментов 9, поэтому вам необходимо совместимое устройство.