Ноутбук Jupyter использует мой графический процессор, однако Pycharm в виртуальной среде - нет. Как мне настроить среду, в которой я работаю, на ноутбуке? - PullRequest
0 голосов
/ 18 февраля 2020

Как видно из названия. Я установил инструментарий CUDA, сделал кузницу исходных текстов из tenorflow gpu. И это прекрасно работает в моем ноутбуке Jupyter. Я бегу CNN, и это занимает около 10 секунд каждую эпоху. Я делаю то же самое для своей Pycharm IDE и получаю сообщение об ошибке:

 File "C:\Users\StackoverflowUser\Anaconda3\envs\Working_Gpu_Environment\lib\site-packages\tensorflow_core\python\client\session.py", line 1472, in __call__
    run_metadata_ptr)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]
     [[metrics/acc/Mean/_127]]
  (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]
0 successful operations.
0 derived errors ignored.

Я думаю, что это должно быть связано со средой, которую я использую, особенно потому, что последний файл, из которого вызвана ошибка из патча, где хранится моя виртуальная среда. Тем не менее, мой недостаток опыта сдерживает меня, я пытался найти среду по умолчанию, которую использует мой ноутбук, но не мог понять, как это сделать. Есть ли другие способы?

...