Нужно ли устанавливать драйвер CUDA для tenorflow-gpu вручную, если я устанавливаю tf через conda?

Question

Я следовал этому руководству и установил tf-gpu, используя conda (https://www.pugetsystems.com/labs/hpc/The-Best-Way-to-Install-TensorFlow-with-GPU-Support-on-Windows-10-Without-Installing-CUDA-1187/)), и это сработало, потому что я вижу «... gpu: 0» в моем распечатанном журнале. До того, как я сделал установку, я уже у меня установлен драйвер CUDA, поэтому я не уверен.

Мне кажется, что conda install tenorflow-gpu поставляется с набором инструментов cuda, cuDNN и т. Д. Мне было интересно, если установка драйвера CUDA является обязательным шагом. В другом сообщении, которое я обнаружил, также не упоминался драйвер (https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc). Но в официальном руководстве по графическим процессорам сказано, что это необходимо, поэтому я в замешательстве. Я делаю это в Windows 10.

Answer 1

По моему опыту вам не нужно устанавливать cuda или cudnn. Достаточно только вашего графического драйвера.

Но в зависимости от вашей системы она может быть не оптимизирована. Для этого вам нужно скомпилировать тензорный поток с нуля и оптимизировать его для вашей системы.