get the CUDA and CUDNN version on windows with Anaconda installe

后端 未结 3 1520
遇见更好的自我
遇见更好的自我 2020-12-16 05:44

There is a tensorflow-gpu version installed on Windows using Anaconda, how to check the CUDA and CUDNN version of it? Thanks.

相关标签:
3条回答
  • 2020-12-16 06:21

    You could also run conda list from the anaconda command line:

    conda list cudnn
    
    # packages in environment at C:\Anaconda2:
    #
    # Name                    Version                   Build  Channel
    cudnn                     6.0                           0
    
    0 讨论(0)
  • 2020-12-16 06:24

    Use the following command to check CUDA installation by Conda:

    conda list cudatoolkit
    

    And the following command to check CUDNN version installed by conda:

    conda list cudnn
    

    If you want to install/update CUDA and CUDNN through CONDA, please use the following commands:

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

    Alternatively you can use following commands to check CUDA installation:

    nvidia-smi
    

    OR

    nvcc --version
    
    0 讨论(0)
  • 2020-12-16 06:35

    Although not a public documented API, you can currently access it like this:

    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
    
    0 讨论(0)
提交回复
热议问题