安装GPU版tensorflow的踩坑之旅
近日来按老师要求,安装gpu版TensorFlow,过程十分艰辛曲折,现将自己经历写出望看者不再踩坑
坑一 CUDA、cudnn、tensorflow版本的匹配
一开始我安装时只是随意的找出一个教程就开始了,有一些博主可能忽略了,没有特意强调对应版本,所以安装完毕后在pycharm中也无法运行,会报错
couldn’t open CUDA library cupti64_100.dll,后边给出一个链接,让我下载cuda10.0,这是坑一
坑二 命令行输入nvidia-smi提示不是内部外部命令
在查找错误的过程中,我想用nvidia-smi来查看一下当前GPU的状态,但是居然报错,先说我不是管理员,后边以管理员身份运行cmd命令窗口时,说这个nvidia不是内部外部命令
原因可能有2种,一是没有填写相关环境变量;二是自己的目录下根本没有nvidia这个可执行程序
检查C:\Program Files\NVIDIA Corporation下NVSMI文件夹是否存在
若没有,可自行下载安装
在安装CUDA10.0以及将其匹配的cudnn也安装后,NVISMI 提取码:bkgo
下载解压后放到上述的路径,即C:\Program Files\NVIDIA Corporation
当然自行安装的这里也是要配置一下环境变量的,同时如果是有此文件夹却不能运行的也是从这里开始,即配置环境,加上安装路径即可
nvidia-smi命令即可运行成功
坑三 提示没有显卡驱动
在解决了没有nvidia-smi后,这个命令倒是可以运行了,也出现熟悉的界面
再回到pycharm进行尝试,,继续报错
错误是
CUDA driver version is insufficient for CUDA runtime version
翻译大概就是说CUDA driver 与 CUDA运行版本不匹配
先看一下自己的各个版本
查看CUDA版本
输入nvcc -V
查看当前显卡驱动版本
输入nvidia-smi
再看一下自己的显卡型号
nvidia-smi -L
上官网查询可知,CUDA10.0要求的是要高于411.31的显卡驱动,而我只有391.25,所以不匹配
然后先去检测一下自己的显卡能下载什么驱动,官网检测
把自己的显卡信息都输入进去,然后搜索
下载后直接安装就可以,以前的版本它在自己安装的过程中会自动卸载,不用担心
安装完毕后:再次查看
再会Pycharm运行!成功
结语
希望大家都可以顺利安装运行,我这里真的是层层报错,我太难了