window 10 安装paddlepaddle 1.7 GPU版本

window 10 安装paddlepaddle 1.7 GPU版本

1)更新显卡驱动
2)安装cuda 10
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
3)安装对应cuda 10 的cudnn包(需要注册)
https://developer.nvidia.com/rdp/cudnn-download
4)将cudnn 解压缩并按照以下指导,拷贝到cuda10对应的文件夹下
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/#download-windows
5)安装anaconda ,建立虚拟环境
6)在虚拟环境下运行以下语句安装paddle 1.7gpu版本
python -m pip install paddlepaddle-gpu==1.7.1.post107 -i https://mirror.baidu.com/pypi/simple

安装完毕后进入python环境,运行以下三条语句测试是否安装成功
python
import paddle.fluid
paddle.fluid.install_check.run_check()

猜你喜欢

转载自www.cnblogs.com/supercodeing/p/12619972.html
今日推荐