Проблема с функцией torch.cuda () - PullRequest
0 голосов
/ 28 мая 2018

Я новичок в факеле.Это работает, когда я использую gpu, запустив мою программу с TensorFlow.Но это проблема с факелом.Я много ищу, но не могу найти полезного ответа.Кто может мне помочь?

Ошибка: enter image description here Моя среда:

  • windows10 64bit
  • python3.6
  • cuda9.0
  • cudnn64
  • gpu: GTX965m

1 Ответ

0 голосов
/ 28 мая 2018

Как вы можете убедиться в примечаниях к выпуску здесь , разработчики Torch решили отказаться от вычислительных возможностей 3.0 и 5.0 устройств из своих сборок, начиная с Torch 0.3.1.

Ваше устройство является устройством Compute 5.0 и поэтому не поддерживается в самых последних версиях Torch.

Вы можете прочитать о том, какие альтернативы вам открыты здесь (в основном используйте более старую версию Torch или создайте свою собственную из исходников с поддержкой вашего графического процессора).

...