cuda11+cudnn8:https://blog.youkuaiyun.com/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百度以及)