Запуск кода Python на CUDA

Question

При попытке запустить этот код https://wltrimbl.github.io/2014-06-10-spelman/intermediate/python/04-multiprocessing.html в моей системе GPU, которая имеет 300 ядер, я использовал комментарий с tf.device ('/ GPU: 0') в начале кода.Но обнаружил, что он не работает на GPU.Тогда я попробовал

import tensorflow as tf
tf.device('/GPU:0'): # for run in GPU
         init = tf.initialize_all_variables()
          # initializing all variables
           sess = tf.Session(
            config=tf.ConfigProto(
             intra_op_parallelism_threads=1))

Этот код работает в графическом процессоре?или есть какой-либо метод для запуска кода Python на GPU.

Answer 1

Нет, он не будет работать без версии TensorFlow, оптимизированной для графического процессора.

Многопроцессорная обработка Python только для процессора.Доступен графический процессор TensorFlow (см. Здесь https://www.nvidia.com/en-us/data-center/gpu-accelerated-applications/tensorflow/). В реализации графического процессора Tensorflow используется CUDA с cuDNN под капотом.

Чтобы запустить собственный скрипт Python на графическом процессоре, вам необходимо использовать библиотеку, такую ​​как PyCUDA илиCupy, которые также используют CUDA API под капотом.