1.首先,查看自己当前的cuda版本和cudnn版本,打开cmd
nvcc --version
我的结果如下。这代表了已经安装了 NVIDIA 的 CUDA 编译器驱动程序,版本为 11.1
2.检查我的cudnn版本。
现存的方法大差不差,查看安装目录里面的一个文件然后用文本模式查看就行。我的是cuda8.9.7
3.查看TensorFlow官网文件,我的TensorFlow-gpu版本为2.4,cuda11.1,而cudnn版本显得有点过高,不知道是否需要调整
1.首先,查看自己当前的cuda版本和cudnn版本,打开cmd
nvcc --version
我的结果如下。这代表了已经安装了 NVIDIA 的 CUDA 编译器驱动程序,版本为 11.1
2.检查我的cudnn版本。
现存的方法大差不差,查看安装目录里面的一个文件然后用文本模式查看就行。我的是cuda8.9.7
3.查看TensorFlow官网文件,我的TensorFlow-gpu版本为2.4,cuda11.1,而cudnn版本显得有点过高,不知道是否需要调整