在自己的GPU上使用CUDA加速跑通paddlenlp项目(环境搭建以及跑通过程中一些遇到的问题)

本文记录了在Ubuntu18.04系统上安装CUDA 11.0、cuDNN以及GPU版PaddlePaddle的过程,包括解决安装过程中遇到的版本不匹配、Python版本过期等问题,最终成功运行Paddlenlp的文本匹配任务。
摘要由CSDN通过智能技术生成

        最近在用百度的paddlepadle框架跑一个文本语义匹配任务,原本我服务器的cuda瓦特掉了所以一直是使用服务器的cpu跑(不用百度的飞桨平台跑是1是因为GPU免费算力时长有限,2是因为这个项目不需要可视化没必要用它的平台),发现有点慢于是决定重装服务器系统(同学帮忙)重新下载cuda进行加速训练。 

服务器系统版本:Linux Ubuntu18.04版本   

首先是通过用户名、密码以及ip用Git Bash连接到服务器,之后安装Anaconda这里就不再赘述。

   1.在Linux下安装GPU版本的PaddlePaddle

 环境要求:python必须是3.6以上,pip必须是20.2.2以上,与Ubuntu18.04版本相对应我此处是选用下载的CUDA11.0 的PaddlePaddle,linux的命令行如下:

python -m pip install paddlepaddle-gpu==2.1.2.post110 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

但此时我还没有下载CUDA,所以报错,安装PaddlePaddle并不成功

 2. 下载安装CUDA

https://developer.nvidia.com/cuda-toolkit-archive中找到对应的CUDA版本

(这个官网需要注册登录)

 

 运行下列命令就可以了,重启服务器之后验证CUDA是否安装成功:用nvidia-smi命令

 可以看到右上角CUDA Version:11.4 ,CUDA安装成功!

再检查cu

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值