Ubuntu16.04 安装paddlepaddle-gpu-2.1.0.post112

最近在研究Paddlepaddle这个深度学习开源框架,发现安装就是个问题,显卡、CUDA版本适配不同版本的paddlepaddle,安装过程中遇到一个问题,在这里做个记录。

CUDA11.2 安装paddlepaddle-gpu

安装版本不对会报错ModuleNotFoundError: No module named 'paddle.io'
使用下边这个安装指令就好啦

python -m pip install paddlepaddle-gpu==2.1.0.post112 -f https://paddlepaddle.org.cn/whl/mkl/stable.html

网上很少有解决这个问题的答案,其实按照官方文档的步骤安装就可以避免很多问题,官方文档yyds。
附上链接

你可能感兴趣的:(paddlepaddle)