1 安装简介
1.1 安装注意事项
显卡型号:RTX4090
驱动版本:550.54.14
宿主机cuda版本:12.4
安装方式:conda
- 注意cuda和cudnn的搭配
最初安装是为了使用PaddleOCR,根据官网提示需要安装cuda和cudnn。这里最关键的就是针对RTX40系的显卡以及对应的550驱动,其对应的版本搭配比较难找。
⚠️
cuda:建议不要使用cuda11.x的版本,需要cuda12.0及以上;
cudnn:建议不要使用cudnn9.x以上版本;
最终安装版本:
cuda == 12.0 cudnn == 8.8.1.3
cuda及cudnn安装推荐链接
-
paddlepadle-gpu安装
当cuda和cudnn安装后,需要安装paddlepaddle,此时有两个注意事项:1) 虽然是在conda环境中,但尽量使用pip install
而非conda install
安装paddlepaddle,否则会对cuda的版本有强制性要求(必须12.6)而非兼容性要求(≥12.0)。2)选择安装cuda12.0以上的paddle即可。且推荐使用paddle3.0,否则在推理时可能会有未知的错误。推荐的安装命令如下:
python -m pip install paddlepaddle-gpu==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
-
可能出现的错误
1) glibc错误:在百度可直接搜索得到,一般是软链接问题,需要替换软链接。⚠️在直接替换软链接前,请注意进行备份或记录。
2)安装paddle3.0时若报错类似cuda-runtime-12.6 not found
的错误,一般是使用conda安装导致的,需要使用pip
进行安装。