1、首先,确定自己的显卡驱动版本最大支持到哪里
比如这里最高支持CUDA11.6,就可以下载CUDA11.6以下的CUDA。这里统一用CUDA10.2+cudnn7.6.5(有需要我打包上传)
2、去CUDA官网下载
3、cudnn要稍微科学上网才方便注册下载
cuDNN Archive | NVIDIA DeveloperExplore and download past releases from cuDNN GPU-accelerated primitive library for deep neural networks.https://developer.nvidia.com/rdp/cudnn-archive#a-collapse765-102解压并复制对应的bin、lib、include在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2的对应文件上
4、验证是否安装成功
打开cmd
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
然后 bandwidthTest.exe 和 deviceQuery.exe,都显示pass就可以了
5、然后创建一个虚拟环境,名字随便(我取了paddle)
然后去paddle的官网查看安装命令(注意是CUDA10.2)
python -m pip install paddlepaddle-gpu==2.3.1 -i https://mirror.baidu.com/pypi/simple
6、可安装官网验证