В ядре Kaggle при выборе опции GPU при проверке torch.cuda.is_available () он говорит, что недоступен - PullRequest
Новая
       29

В ядре Kaggle при выборе опции GPU при проверке torch.cuda.is_available () он говорит, что недоступен

1 голос
/ 30 марта 2019

Я создал ядро ​​для законченного соревнования Kaggle, в котором я использовал pytorch. При проверке наличия cuda возвращает False.

Я проверил опцию GPU из настроек, и она говорит, что она включена в нижней панели с информацией о ресурсах. Я попытался перезапустить сеанс без каких-либо изменений. В чем может быть проблема? (возможно, установлен только процессор?)

Ответы [ 2 ]

0 голосов
/ 30 марта 2019

Проблема была с выбранной конфигурацией докера из настроек.Выбор «Последние доступные» исправил проблему.

0 голосов
/ 30 марта 2019

Запустите ядро ​​и turn-on опцию GPU в настройках на правой панели управления.Затем выполните следующие команды. 

    [1]:import torch
    [2]:torch.cuda.device_count()
    Out[2]:1

    [3]:torch.cuda.is_available()
    Out[3]:True
...