显卡3050ti等安培架构的GPU安装paddlepaddle

3050ti等安培架构的GPU安装paddlepaddle

官网原话

如果您使用的是安培架构的GPU，推荐使用CUDA11.2
如果您使用的是非安培架构的GPU，推荐使用CUDA10.2，性能更优

其他配套

确定软件版本后，到官网下载对应版本的软件
CUDN：CUDA Toolkit Archive | NVIDIA Developer

cuDNN：cuDNN Archive | NVIDIA Developer

这里需要下载CUDA11.2 cuDNN8.1
安装paddlepaddle命令：python -m pip install paddlepaddle-gpu==2.2.2.post112 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

一切顺利