web-dev-qa-db-ja.com

AnacondaがインストールされているWindowsでCUDAおよびCUDNNバージョンを取得する

Anacondaを使用してWindowsにインストールされているtensorflow-gpuバージョンがあります。そのCUDAおよびCUDNNバージョンを確認するにはどうすればよいですか？ありがとう。

pythontensorflowanacondagpucudnn
3
user297850

また、anacondaコマンドラインからconda listを実行することもできます。

conda list cudnn

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

次のコマンドを使用して、CondaによるCUDAのインストールを確認します。

conda list cudatoolkit

そして、condaによってインストールされたCUDNNバージョンを確認する次のコマンド：

conda list cudnn

CONDAを介してCUDAおよびCUDNNをインストール/更新する場合は、次のコマンドを使用してください。

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

または、次のコマンドを使用して、CUDAのインストールを確認できます。

nvidia-smi

OR

nvcc --version
6
sidk

ドキュメント化された公開APIではありませんが、現在は次のようにしてアクセスできます。

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