Ускорение GPU Numba на & ldquo; Высокопроизводительный GPU & rdquo; вместо интегрированных? Python-3.x [дубликат]

Выполните следующие шаги:

  1. Откройте файл catalina.sh из tomcat / bin.
  2. Chnage JAVA_OPTS в
    JAVA_OPTS="-Djava.awt.headless=true -Dfile.encoding=UTF-8 -server -Xms1536m 
    -Xmx1536m -XX:NewSize=256m -XX:MaxNewSize=256m -XX:PermSize=256m 
    -XX:MaxPermSize=256m -XX:+DisableExplicitGC"
    
  3. Перезапустите ваш tomcat
36
задан Tim 18 June 2016 в 05:55
поделиться

2 ответа

Вы можете установить переменные среды в записной книжке с помощью os.environ. Перед инициализацией TensorFlow выполните следующее:

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

Вы можете дважды проверить, что у вас есть правильные устройства, видимые для TF

from tensorflow.python.client import device_lib
print device_lib.list_local_devices()

Я склонен использовать это из служебного модуля, такого как notebook_util

import notebook_util
notebook_util.pick_gpu_lowest_memory()
import tensorflow as tf
80
ответ дан Yaroslav Bulatov 18 August 2018 в 05:41
поделиться
  • 1
    Я взял на себя смелость обобщить (и процитировать) различные ответы (включая ваши) в некоторых примерах документации stackoverflow.com/documentation/tensorflow/10621 Надеюсь, вы не против. Не стесняйтесь редактировать его. – npf 24 June 2017 в 13:34

Вы можете сделать это быстрее без импорта только с помощью magics :

%env CUDA_DEVICE_ORDER=PCI_BUS_ID
%env CUDA_VISIBLE_DEVICES=0

Обратите внимание, что все переменные env являются строками, поэтому нет необходимости использовать ". Вы можете проверить, что env-variable настроена путем запуска: %env <name_of_var>. Или проверьте все из них с помощью %env.

16
ответ дан Salvador Dali 18 August 2018 в 05:41
поделиться
  • 1
    Этот параметр env влияет на другие скрипты python? поскольку это похоже на изменение переменной окружения os. – JenkinsY 16 May 2018 в 11:21
  • 2
    @JenkinsY вы можете легко проверить это, установив переменную в один скрипт и прочитав ее в другом. – Salvador Dali 23 May 2018 в 07:35
Другие вопросы по тегам:

Похожие вопросы: