tensorflow和cuda以及cudnn版本对齐

时间: 2023-08-02 admin IT培训

tensorflow和cuda以及cudnn版本对齐

tensorflow和cuda以及cudnn版本对齐

起因

今天想用下买了已经很久的GPU来玩tensorflow,毕竟显卡也不能只是用于玩游戏吧。然后发现因为cuda 和cudnn版本问题要卸载掉旧的重装新的,踩了一天的坑,来记录一下。

cuda版本问题

在NVIFIA CONTROL PANEL里显示,我应该装cuda 11.2版本,但是官网上最新版本是11.1。这就有点坑人了。cuda 11.2的download网址如下:cuda 11.2

登录NVIDIA网站问题

登录NVIDIA网站的时候,验证码的问题老是被拦截,但是无法登录,解决方法有两个:
1.不要开任何代理软件
2.用UU加速器可以(是的你没听错UU加速器)

NVSMI文件夹丢失问题

在C:\Program Files\NVIDIA Corporation文件夹下没有NVSMI文件夹,这个时候不要去下载NVSMI文件,如果下载了NVSMI文件放在C:\Program Files\NVIDIA Corporation中,运行nvidia-smi会显示

Failed to initialize NVML: Unknown Error

其实nvidia-smi在
C:\Windows\System32\DriverStore\FileRepository\nvdm*\nvidia-smi.exe
这个目录下,进入对应目录运行nivdia-smi就可以了

cudnn版本问题

CUDA11.2 对应的cudnn版本与CUDA11.1对应的cudnn版本一致,亲测。