web-dev-qa-db-fra.com

Nvidia Cudatoolkit contre Conda Cudatoolkit

Jusqu'à ce jour, j'ai utilisé Tensorflow-GPU en l'installant en utilisant pip et le logiciel associé à Cuda et les logiciels/pilotes Nvidia du site Web de Nvidia . Récemment, j'ai découvert qu'en utilisant conda install tensorflow-gpu installe également cudatoolkit et cudnn .

Alors, en quoi ceux-ci (ceux fournis par conda) sont-ils différents de ceux que j'ai téléchargés sur le site Web de Nvidia?

Dans mon premier (précédent) environnement, conda list a montré que je n'avais installé que TensorFlow (de PyPi) et pas de cudnn/cudatoolkit, mais tout fonctionnait toujours. This is my base environment where I installed Tensorflow-GPU using pip

Aussi, dans un nouvel environnement dans lequel j'ai exécuté conda install tensorflow-gpu, conda list m'a montré que tensorflow-gpu a été installé avec cudatoolkit et cudnn par Anaconda . Et dans cet environnement aussi, tout fonctionnait bien. This is a new environment where I used conda to install tensorflow-gpu

Cela signifie-t-il que le téléchargement et l'installation de Cuda à partir du site Web de Nvidia sont uniquement nécessaires si j'utilise pip pour installer TensorFlow?

pythontensorflowanacondaconda
6
Rajdeep Dutta

Si vous utilisez anaconda pour installer tensorflow-gpu, oui, il installera cuda et cudnn pour vous dans le même environnement conda que tensorflow-gpu. Tout ce dont vous avez besoin pour vous installer est le dernier pilote nvidia (pour qu'il fonctionne avec le dernier niveau CUDA et tous les anciens niveaux CUDA que vous utilisez.)

Cela présente de nombreux avantages par rapport à la méthode pip install tensorflow-gpu:

  1. Anaconda installera toujours la version CUDA et CuDNN que le code TensorFlow a été compilé pour utiliser.
  2. Vous pouvez avoir plusieurs environnements conda avec différents niveaux de TensorFlow, CUDA et CuDNN et utiliser simplement conda activate pour basculer entre eux.
  3. Vous n'avez pas à vous occuper de l'installation manuelle de CUDA et cuDNN au niveau du système.

L'inconvénient par rapport à pip install tensorflow-gpu, est que la dernière version de tensorflow est ajoutée à pypi des semaines avant qu'Anaconda ne puisse mettre à jour la recette de conda et publier ses versions de la dernière version de TensorFlow.

1
William D. Irons