PaddleNLP icon indicating copy to clipboard operation
PaddleNLP copied to clipboard

如何制作在不同CUDA版本的镜像?

Open Ywandung-Lyou opened this issue 2 years ago • 2 comments

需要部署的设备使用CUDA 11.0,可是提供的镜像registry.baidubce.com/paddlepaddle/paddlenlp:2.4.0-gpu-cuda10.2-cudnn7要求的版本是10.2。请问有提供不同CUDA版本的镜像吗?

Ywandung-Lyou avatar Sep 17 '22 09:09 Ywandung-Lyou

cuda 11.2版本的pipelines计划会提供,但时间没那么快,需要提交给qa测试:

如果想自己编译一个镜像,请拉取官方的cuda 11.2的镜像

nvidia-docker pull registry.baidubce.com/paddlepaddle/paddle:2.3.2-gpu-cuda11.2-cudnn8

将paddle的源码放进去,然后参考: https://github.com/PaddlePaddle/PaddleNLP/tree/develop/pipelines/docker

w5688414 avatar Sep 17 '22 10:09 w5688414

Please refer to this pr for compiling you own cuda version's Docker. https://github.com/PaddlePaddle/PaddleNLP/pull/3315

w5688414 avatar Sep 20 '22 03:09 w5688414

This issue is stale because it has been open for 60 days with no activity. 当前issue 60天内无活动,被标记为stale。

github-actions[bot] avatar Dec 07 '22 07:12 github-actions[bot]

This issue was closed because it has been inactive for 14 days since being marked as stale. 当前issue 被标记为stale已有14天,即将关闭。

github-actions[bot] avatar Dec 21 '22 11:12 github-actions[bot]