惨痛史:
自己一开始由于没有什么了解,就随便网上找了一个教程开始弄,也没有在意是否匹配自己的显卡驱动情况,就安装了,最开始受罪的是我的Ubuntu,哎,这个就不提了,都是泪,之后再在windows10 上安装cuda、cudnn等也是一头雾水,各种版本问题啊、下载问题啊、虚拟环境问题啊、BERT支持问题啊,导致按了又重装一遍,一开始安装的时候NVIDIA驱动都被我搞没了,安装完成之后用几天等到不匹配的时候才发现这个问题,心态都有点要崩了(像我这么积极的人 哈哈 ),昨天开始全部重新安装,顺便写下全部过程,应该是踩了不少坑,希望可以对后来人有所帮助。
写在开头
大家先要了解自己的驱动支持版本,我是在昨天更新驱动之后才可以支持cuda10.1的,要不然之前一直最高只支持9.1版本的,那么如何查看本地驱动支持的CUDA版本呢?
操作:打开NVIDIA驱动程序,选择导航栏帮助-系统信息,选择组件,查看动态链接库文件NVCUDA.DLL 后面会写最高支持的CUDA版本哦
这里再新增一点(菜鸟的我又踩到坑了。。。):虽然驱动支持cuda10.1的,但是后面在跑模型的时候出现本机的算力不足的问题,也就是硬件无法支持cuda的高版本,最后我卸载cuda10.1的,然后重装了cuda9.0的,这个问题就解决了,笔记本还是有点菜啊(920M 对应算力3.5)
这里附一张显卡算力表:
下面带一张cuda对应需求算力:
下载对应CUDA版本
这里由于官网找得比较麻烦,大家可以通过点击下面链接找到较为全面的CUDA版本(链接地址是:https://developer.nvidia.com/cuda-toolkit-archive)
CUDA各版本官方下载地址
安装CUDA
下载之后就安装吧
这里顺便提一下,因为CUD