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
測試是否安裝成功
- 在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()