Which TensorFlow and CUDA version combinations are compatible?

后端 未结 6 791
慢半拍i
慢半拍i 2020-11-22 13:07

I have noticed that some newer TensorFlow versions are incompatible with older CUDA and cuDNN versions. Does an overview of the compatible versions or even a list of officia

6条回答
  •  春和景丽
    2020-11-22 13:32

    if you are coding in jupyter notebook, and want to check which cuda version tf is using, run the follow command directly into jupyter cell:

    !conda list cudatoolkit
    
    !conda list cudnn
    

    and to check if the gpu is visible to tf:

    tf.test.is_gpu_available(
        cuda_only=False, min_cuda_compute_capability=None
    )
    

提交回复
热议问题