最近项目用到paddlenlp,并且需要对模型进行微调,自己电脑的cpu根本带不动,于是部署了gpu版本的云服务器,第一次接触gpu,耗时一个下午终于弄明白了,在此记录下来,希望对大家有所帮助。
文章目录
安装Paddlepaddle
安装Paddlenlp之前首先要安装Paddlepaddle,这是被很多同学忽略的一点。我最开始也是直接安装Paddlenlp,在import的时候就一直出错,最后才发现官方文档最开始就强调了“以下安装过程默认用户已安装好paddlepaddle-gpu或paddlepaddle(版本大于或等于2.0)”。
Paddlepaddle官方安装说明已经非常详细,但是有些细节容易被忽略,下面着重强调。
CUDA版本确认及安装
Paddlepaddle目前GPU版本支持CUDA 10.1/10.2/11.0/11.1/11.2,所以一定一定要看弄清楚云服务器上是否已经安装了CUDA以及CUDA的版本。
查看NVIDIA支持的最高版本
点击桌面NAVIDIA控制面板图标——>系统信息——>组件,可以看到NVIDIA支持的最高CUDA版本