Попытка использовать Cuda в pyTorch

Question

Я работал над PyTorch и хотел использовать тензоры Cuda, но мне было трудно заставить его работать.

Когда я использую строку torch.cuda.is_available (), он возвращает false .

Но когда я использую ту же строку в командной строке anaconda, она возвращает true. Что дает? Нужно ли как-то настраивать устройство? Или, может быть, переводчик включает мой графический процессор? Все, что я хочу, - это чтобы мой графический процессор распознавался как пригодный для использования в CUDA и мог использоваться в коде.

Я использую:

python 3.7

CUDA 10.1

GPU: GTX 1070

Answer 1

Я обнаружил проблему, мой переводчик ошибся, и я изменил ее, чтобы она соответствовала моему интерпретатору подсказок анаконды. Работает как шарм