一些从源码编译 pytorch的时候的问题

在尝试从源码编译PyTorch时遇到了cudnn版本错误,编译版本与使用版本不一致,导致无法在两块k80 GPU上运行。通过检查发现cudnn_v7头文件实为v5版本。解决方法包括删除旧的cudnn路径,使用正确版本的cudnn重新编译。在PyCharm中远程调试时也遇到了问题,通过复制库文件到anaconda环境解决了导入错误。确保使用源码安装PyTorch时,cudnn已正确配置。
摘要由CSDN通过智能技术生成

由于遇到了不能使用两块 k80GPU,
和pycharm 远程连接服务器的时候不能调试的问题:

ImportError: dlopen: cannot load any more object with static TLS

然后 conda install 的 pytorch 使用的 cudnn 版本的6的,使用
torch.backends.cudnn.version()来查看使用的 cunn版本。
详细参见我提的 issue cudnn version found error, v7 files but got v5.1.10 #3896.

所以需要从源码编译中安装,然后遇到一堆坑。

1. 编译的时候cudnn版本不对

使用的 cuda 8.0, 在编译的时候提示
note: #pragma message: CuDNN v5 found, but need at least CuDNN v6.
在 log中找到
-- Found cuDNN: v5.1.10 (include: /usr/local/cuda-8.0/include, library: /usr/lib/x86_64-lin

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值