在Pycharm中安装Paddle深度学习框架安装(GPU版本)

前提环境 python3.7,在anaconda虚拟环境下进行

首先需要确认的是电脑显卡最高所能支持的CUDA版本,支持的CUDA版本低,则不能安装高版本的CUDA工具包和加速计算库cudnn。支持的CUDA版本高,能安装当前版本以下的所有版本CUDA工具包和加速计算库cudnn。查看电脑NVIDIA显卡当前驱动程序所支持的CUDA版本,在Windows桌面下鼠标右击,选中NVIDIA控制面板,点击打开控制面板左下角的系统信息,点击‘显示’右边的‘组件’,看到NVCUDA64.DLL属性信息 CUDA后面的 ‘11.3.70’ 即为NVIDIA显卡当前驱动程序所支持的最高CUDA版本。
在这里插入图片描述
通过查看https://developer.nvidia.com/rdp/cudnn-archive找到对应的CUDA版本和cuDNN版本
找版本方法我的NVIDIA CUDA版本9.1.84 因此我需要找≤9.1
在这里插入图片描述

#安装CUDA工具包
conda install cudatoolkit=9.0
###安装cudnn加速计算库
conda install cudnn=7.6.4
#如果您的机器安装的是CUDA9,请运行以下命令安装
python -m pip install paddlepaddle-gpu==1.8.4.post97 -i https://mirror.baidu.com/pypi/simple

通过以上方法就可以成功安装GPU
在这里插入图片描述

  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值