1 gpu实例
CentOS 7.6
选择自动安装cuda,勾选自动 CUDA 版本 10.1.168 / Driver 版本 440.64.00 / CUDNN 版本 7.6.5
2 切换源
curl -o /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo yum clean all yum makecache
3 安装Python、paddle
yum install python3-pip python3 -m pip install paddlepaddle-gpu==1.8.3.post107 -i https://mirror.baidu.com/pypi/simple python3 -m pip install paddlehub -i https://mirror.baidu.com/pypi/simple
4 需要执行下
export CUDA_VISIBLE_DEVICES=0
不过我用设备信息查到的id应该是1,比较玄学