最近跑别人论文里的代码,要用到BERT模型,但是本人没有.深度学习神经网络方面的基础,刚开始模型什么都看不懂,后来看一点神经网络的一些基础知识点,也没看完就看不下去了,急着想把代码跑一跑,然后就死在第一步了,安装环境!首先说下我的情况,我的代码里本来有import torch,我刚开始不知道这个torch是什么,就以为普通的包,就直接在pycharm引入,果不其然失败了,我就去网上搜了一个下载方式,后来我才知道我下载的是cpu版本的pytorch 1.10。本以为下载好了,我就去运行代码了,结果预训练也需要下载模型和数据集,我真的什么都不知道我就又去下载预训练模型需要的东西,终于下好了以为这下可以跑代码了吧,结果报错,就是下图:
我就一脸懵又去搜了下这个错误,我才发现,一个模型能跑起来要依赖的东西还挺多的,我是真滴菜!
pytorch、显卡、显卡驱动、cuda版本是如何对应的 - 简书 (jianshu.com)
然后我就去搜了博客按着博客一步一步安装CUDA和cudnn,参考博客如下: