web-dev-qa-db-fra.com

obtenir la version CUDA et CUDNN sur Windows avec Anaconda installé

Il existe une version tensorflow-gpu installée sur Windows à l'aide d'Anaconda, comment en vérifier la version CUDA et CUDNN? Merci.

pythontensorflowanacondagpucudnn
3
user297850

Vous pouvez également exécuter conda list à partir de la ligne de commande anaconda:

conda list cudnn

# packages in environment at C:\Anaconda2:
#
# Name                    Version                   Build  Channel
cudnn                     6.0                           0
6
John H

Utilisez la commande suivante pour vérifier l'installation de CUDA par Conda:

conda list cudatoolkit

Et la commande suivante pour vérifier la version de CUDNN installée par conda:

conda list cudnn

Si vous souhaitez installer/mettre à jour CUDA et CUDNN via CONDA, veuillez utiliser les commandes suivantes:

conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn

Vous pouvez également utiliser les commandes suivantes pour vérifier l'installation de CUDA:

nvidia-smi

OR

nvcc --version
6
sidk

Bien qu'il ne s'agisse pas d'une API publique documentée, vous pouvez actuellement y accéder comme suit:

from tensorflow.python.platform import build_info as tf_build_info
print(tf_build_info.cuda_version_number)
# 9.0 in v1.10.0
print(tf_build_info.cudnn_version_number)
# 7 in v1.10.0
5
jdehesa