安装NCLL2(Ubuntu1804+cuda10.0+cudnn7.6.4)

本文详细指导了如何在Ubuntu 18.04系统上为PaddlePaddle 1.8.5 GPU版本安装所需的NCCL2库,适用于CUDA 10.0和CUDNN 7.6.4环境,针对CUDA 10.1以上版本的缺失提供了解决方案。
摘要由CSDN通过智能技术生成

安装NCCL2(Ubuntu1804+cuda10.0+cudnn7.6.4)

问题描述:安装PaddlePaddle1.8.5的GPU版本,安装命令参考链接https://paddlepaddle.org.cn/install/quick/zh/1.8.5-linux-pip

可以看到:PaddlePaddle1.8.5暂不支持cuda10.1以及cuda10.2的pip和conda安装方式,若需要请自行源码编译查看源码编译步骤

这里我使用的cuda10.0安装,命令是如下:

python -m pip install paddlepaddle-gpu==1.8.5.post107 -i https://mirror.baidu.com/pypi/simple

很快安装好,进行测试:

使用 python 进入python解释器,输入import paddle.fluid ,再输入 paddle.fluid.install_check.run_check()

不通过需要nccl恩,官网的版本找到都cuda10.1及之后的,之前的有点难找,贴个链接方便一下

https://developer.nvidia.com/nccl/nccl-legacy-downloads

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值