Импорт TensorFlow останавливает запуск программы python

Question

У меня есть настройка Python Tools в Visual Studios с установленным CPython.

В Visual Studios, если я выполнить следующий код:

print("hello"); 
import numpy; 
print("hello"); 

Программа прекрасно работает, печатает два 'привет', и выходит нормально.

Однако, если я запускаю следующий код:

print("hello"); 
import tensorflow; 
print("hello"); 

Программа зависает, печатает один «привет», и отказывается продолжать.

Все пакеты должны быть правильно установлены - использование TensorFlow в интерактивном окне Python позволяет печатать правильный результат и отлично работать.

Почему программа зависает во втором сценарии?

Answer 1

После импорта tensorflow он автоматически пытается загрузить CUDA, он печатает что-то вроде этого:

I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally 

Так что я думаю, что происходит в том, что вы не правильно установили CUDA, и это из-за неисправного Это. Вы можете попробовать установить версию CPU, которая не использует графический процессор, и не загружает эти библиотеки.