window 10 安装paddlepaddle 1.7 GPU版本

window 10 安装paddlepaddle 1.7 GPU版本
window 10 安装paddlepaddle 1.7 GPU版本_第1张图片

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
测试是否安装成功

  • 在cmd中使用命令 nvcc -V可查看cuda版本
  • 在cmd命令中nvidia-smi可查看gpu使用情况,如果不能识别命令,需要设置Path变量,我的目录为: C:\Program Files\NVIDIA Corporation\NVSMI

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()

window 10 安装paddlepaddle 1.7 GPU版本_第2张图片

你可能感兴趣的:(window 10 安装paddlepaddle 1.7 GPU版本)