推薦的順序是子寓,查看AI框架梁厉,選擇一個(gè)合適的版本辜羊,根據(jù)AI框架的版本安裝對(duì)應(yīng)的CUDA,根據(jù)CUDA版本安裝cudnn词顾,檢查一下電腦nvidia顯卡驅(qū)動(dòng)的版本是否為最新八秃,是則可以安裝AI框架,否則更新驅(qū)動(dòng)再安裝
1肉盹、安裝顯卡驅(qū)動(dòng)昔驱,并查看驅(qū)動(dòng)最高支持的cuda版本
注意到 paddleGPU 目前最高支持到 cuda11.7
查看顯卡信息 : nvidia-smi
2、安裝cuda
CUDA(Compute Unified Device Architecture)上忍,是顯卡廠商N(yùn)VIDIA推出的運(yùn)算平臺(tái)骤肛。 CUDA?是一種由NVIDIA推出的通用并行計(jì)算架構(gòu),該架構(gòu)使GPU能夠解決復(fù)雜的計(jì)算問(wèn)題窍蓝。
選擇對(duì)應(yīng)的版本下載 https://developer.nvidia.com/cuda-toolkit-archive 安裝
測(cè)試 nvcc -V
3腋颠、安裝cudnn
NVIDIA cuDNN是用于深度神經(jīng)網(wǎng)絡(luò)的GPU加速庫(kù)
選擇對(duì)應(yīng)的版本下載 https://developer.nvidia.com/rdp/cudnn-archive 安裝
配置cuDNN
cuDNN下載完成后,是一個(gè)壓縮包吓笙,解壓完成后淑玫。包含bin,include,lib三個(gè)目錄。
把cuda\bin\cudnn64_8.dll復(fù)制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin 目錄下. 把\cuda\ include\cudnn.h復(fù)制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include 目錄下. 把\cuda\lib\x64\cudnn.lib復(fù)制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64 目錄下.
把cuda\bin中的下面文件復(fù)制到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/
pip 安裝
python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
最后驗(yàn)證一下絮蒿,安裝是否成功。
import paddle.fluid
paddle.fluid.install_check.run_check()
運(yùn)行成功
相關(guān)問(wèn)題
1叁鉴,報(bào)錯(cuò):Could not locate cudnn_ops_infer64_8.dll. Please make sure it is in your library path!
解決:
去官網(wǎng):https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-zlib-windows
下載一個(gè)zlib包歌径,解壓縮后找到zlibwapi.dll文件,剪切到C:\Windows\System32