centos+3090+cuda11+pytorch1.7下安装Warpctc

cuda11+cudnn8:https://blog.csdn.net/qq_33266320/article/details/113186054

pytorch安装:https://pytorch.org/(好像官网截止2021.1.28为止只有支持到cuda11.0,我在cuda11.1下装了下,也能用,但是很多操作跑到GPU了,现在准备源码安装)

            pytorch源码编译cudn8以上会报错,pip install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses

Found CUDNN: /usr/local/cuda-11.1/lib64/libcudnn.so  
-- Found cuDNN: v?  (include: /usr/local/cuda-11.1/include, library: /usr/local/cuda-11.1/lib64/libcudnn.so)
CMake Error at /home/saini/anaconda3/lib/python3.7/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:174 (message):
  PyTorch requires cuDNN 7 and above.

                https://github.com/pytorch/pytorch/issues/41593参考这里,移动version.h文件

更新gcc版本:http://www.vpser.net/manage/centos-6-upgrade-gcc.html   我直接用7.3的可以用

warpctc安装:https://github.com/guanfaqian/warp-ctc(直接git clone这个,不要clone百度以及)

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值