тест скорости для тестирования тензорного потока

Я сомневаюсь, правильно ли настроен тензор на моем поле gpu, так как он приблизительно на 100 раз медленнее на итерацию для обучения простой модели линейной регрессии (batchsize = 32, 1500 входных функций, 150 выходных переменных) на моей модной машине gpu, чем на моем ноутбуке.тест скорости для тестирования тензорного потока

Я использую Titan X с современным процессором и т. Д. Nvidia-smi говорит, что я использую только 10% загрузки gpu, но я ожидаю, что это из-за небольших партийных файлов. Я не использую feed_dict для перемещения данных в граф вычислений. Все идет через tf.decode_csv и tf.train.shuffle_batch.

Есть ли у кого-нибудь рекомендации относительно того, как легко проверить, правильно ли установлена моя установка? Есть ли простые контрольные показатели скорости? Разница в скорости между моим ноутбуком и машиной gpu настолько драматична, что я ожидаю, что все будет неправильно настроено.

источник

2016-02-29 DBelanger

Некоторые примеры не оптимизированы для GPU и может получить 100x медленнее, когда GPU включен из-за копий, вот один протокол HTTPS: //github.com/tensorflow/tensorflow/issues/838 –

Попробуйте tensorflow/tensorflow/models/image/mnist/convolutional.py, что будет печатать за шаг.

На Теслах K40c, которые должны получить около 16 ms за шагом, в то время как о 120 ms для CPU-только на моей 3-летней машине

источник

2016-02-29 21:40:42

Только для записи. Я использую GeForce GTX 960M и получаю время шага с 16.1 до 17.7, но в основном около 16.2. – user728785

Если вы ищете место установки tensorflow (где находится файл), посмотрите здесь: http://stackoverflow.com/questions/33616732/where-is-the-folder-for-installing-tensorflow-with- pip-mac-osx – jrieke

выглядит как convolutional.py, удаленный, вы все еще можете увидеть его в истории github - https://github.com/tensorflow/tensorflow/blob/6431560b7ec3565154cb9cdc9c827db78ccfebe7/tensorflow/models/image/mnist/convolutional. py # L265 –

Расширения Yaroslavs ответа: Вот как сделать все тестирование процесса (CUDA и cudNN уже установлена)

git clone https://github.com/tensorflow/models.git

Создание Virtuel среды для tensorflow и установить tensorflow

virtualenv --system-site-packages -p python3 tf-venv3 
source tf-venv3/bin/activate 
pip install --upgrade pip 
pip install --upgrade tensorflow-gpu

Запустите модель в вашей Virtuel среде

python models/tutorials/image/mnist/convolutional.py

Мой GTX 1070 Потребности ~ 5мс на шаг

источник

2017-06-02 11:14:19

тест скорости для тестирования тензорного потока

ответ

Смежные вопросы