项目场景:
运行sktime-dl需要用到tensorflow-gpu2.6 ,服务器上有conda环境。但是cuda与cudnn版本号出了问题,下面记录一下过程。问题描述:
查看机器信息:
查看cuda版本 11.4
- 方法一 : 在路径下查看 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
- 方法二:
- -nvcc -V
- - v11.4 (管理人员告知)查看tensorflow版本
pip list tensorflow
– 2.6.0
查看cudnn版本 8.2.2 (学姐告知)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include\cudnn.h
另外新版本cudnn要用
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
来查看版本linus下用这个命令
cat /usr/local/cuda-11.4/include/cudnn_verison.h | grep CUDNN_MAJOR -A 2