'TypeError'>: Wrong Type. Поднял документацию, говорит что не может выделить память из видеокарты, кто сталкивался, чем лечить?
cuFFT failed to allocate GPU or CPU memory
Можно код
запустите на http://colab.research.google.com/ если там не заработает - проблема в коде, если заработает, то в вашей ОС там оно проверено и оптимизировано
Код работает через библиотеку cuda, что подразумевает запуск через GPU, собственно Не запустится он там хоть как
Там есть и GPU, и TPU
https://cloud.google.com/tpu
Обсуждают сегодня