У меня проблема при использовании среды conda.
Я создаю среду просто:
conda create -n NAME python==3.5
Итак, существует виртуальная среда Python 3.5.
Затем я вхожу в виртуальную среду и устанавливаю tenorflow-gpu с:
conda install tensorflow-gpu
Теперь я вижу пакет tenorflow-gpu вместе с cudatoolkit и cudnn.
Но когда я запускаю простую программу, она просто не может использовать gpu. Я не думаю, что это проблема или пакет драйвера nvidia из разных ресурсов, потому что он работает в старой среде conda, но когда я создаю новую, это не удается.
Более интересно, что он будет использовать пакет (tenorrt и uff) снаружи окружающей среды.
Когда я создаю новую среду с python 3.6.Он работает должным образом с помощью tenorflow-gpu и не может импортировать пакет, который не установлен в среде.
Кто-нибудь знает, как решить проблему?
Ubuntu 16.04
Tensorflow 1.8.0
Python 3.5 и 3.6
TensorRT 5.0.0.10