一、安裝 PyCharm
PyCharm 官網下載頁面
記得全部勾選。
二、安裝 miniconda
miniconda 官網下載頁面
根據你的操作系統選擇。
記得勾選前三個。
三、安裝 CUDA
首先 nvidia-smi 查看支持最高的 CUDA 版本。
然后去 nvidia 官網下載 CUDA,選擇適合你的版本。
CUDA 官網下載頁面
四、安裝 cudnn
下載 cudnn 需要注冊 nvidia 賬號并登錄。
cudnn 官網下載頁面
安裝后檢查這 4 個是否在系統環境變量 Path 中:
之后,nvcc -V 檢查 CUDA 是否安裝完畢:
最后,去 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\extras\demo_suite,在命令行檢查 bandwidthTest.exe 和 deviceQuery.exe,輸出結果正常說明 CUDA 和 cudnn 安裝成功。
五、安裝 paddlepaddle-gpu 和 paddlex
python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu126/pip install paddlex==3.0.0
六、安裝項目依賴
以飛槳的分割框架為例,克隆項目到本地。
git clone https://github.com/PaddlePaddle/PaddleSeg
最后,安裝項目依賴。
pip install -r requirements.txt
至此,環境搭建完成。