阿里云的gpu实例配置paddlepaddle

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,比较玄学

 

Leave a Reply

Your email address will not be published. Required fields are marked *