美文网首页
windows中paddlegpu 安装教程

windows中paddlegpu 安装教程

作者: hiekay | 来源:发表于2023-06-07 18:47 被阅读0次

    推荐的顺序是,查看AI框架,选择一个合适的版本,根据AI框架的版本安装对应的CUDA,根据CUDA版本安装cudnn,检查一下电脑nvidia显卡驱动的版本是否为最新,是则可以安装AI框架,否则更新驱动再安装

    1、安装显卡驱动,并查看驱动最高支持的cuda版本

    注意到 paddleGPU 目前最高支持到 cuda11.7

    image.png

    查看显卡信息 : nvidia-smi

    image.png

    2、安装cuda

    CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

    选择对应的版本下载 https://developer.nvidia.com/cuda-toolkit-archive 安装

    image.png image.png

    测试 nvcc -V

    image.png

    3、安装cudnn

    NVIDIA cuDNN是用于深度神经网络的GPU加速库

    选择对应的版本下载 https://developer.nvidia.com/rdp/cudnn-archive 安装

    image.png

    配置cuDNN

    cuDNN下载完成后,是一个压缩包,解压完成后。包含bin,include,lib三个目录。

    把cuda\bin\cudnn64_8.dll复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin 目录下. 把\cuda\ include\cudnn.h复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include 目录下. 把\cuda\lib\x64\cudnn.lib复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64 目录下.

    把cuda\bin中的下面文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin 目录下.

    cudnn_adv_infer64_8.dll
    cudnn_adv_train64_8.dll
    cudnn_cnn_infer64_8.dll
    cudnn_cnn_train64_8.dll
    cudnn_ops_infer64_8.dll
    cudnn_ops_train64_8.dll
    

    重启

    4、安装Paddlepaddle-GPU

    下载https://www.paddlepaddle.org.cn/

    image.png

    pip 安装

    python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
    

    最后验证一下,安装是否成功。

    import paddle.fluid
    paddle.fluid.install_check.run_check()
    

    运行成功

    image.png

    相关问题

    1,报错:Could not locate cudnn_ops_infer64_8.dll. Please make sure it is in your library path!


    image.png

    解决:

    去官网:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-zlib-windows

    下载一个zlib包,解压缩后找到zlibwapi.dll文件,剪切到C:\Windows\System32

    image.png

    相关文章

      网友评论

          本文标题:windows中paddlegpu 安装教程

          本文链接:https://www.haomeiwen.com/subject/bzqoedtx.html