apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 09:31:35 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Serving]Support 11.2 CUDA (#1976)

Browse Source 
* add workspace_size option

* support 11.2 cuda

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
WJJ1995
2023-05-23 15:16:07 +08:00
committed by GitHub
parent 3e7cb88049
commit ee81605e5d
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
serving/Dockerfile Normal file → Executable file
View File

@@ -13,7 +13,7 @@
# limitations under the License.
FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
ARG http_proxy
ARG https_proxy

2
serving/scripts/build.sh Normal file → Executable file
View File

@@ -86,7 +86,7 @@ nvidia-docker run -i --rm --name ${docker_name} \
-e "http_proxy=${http_proxy}" \
-e "https_proxy=${https_proxy}" \
-e "trt_version=${trt_version}"\
nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04 \
nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 \
bash -c \
'export https_proxy_tmp=${https_proxy}
export http_proxy_tmp=${http_proxy}