安装完cuda之后 安装cudnn


nvcc --version(自己安装的cuda版本) 版本需要和nvidia-smi查看到的cuda版本(nvidia驱动对应的cuda版本)一致

1、解压:tar zxvf cudnn-10.1-linux-x64-v7.5.0.56.tgz

include lib64两个文件夹

2、cuda安装目录为/usr/loca/cuda-10.1

把cudnn的include/ 和 lib64/目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64里面

$ sudo cp -P include/cudnn.h /usr/local/cuda-10.1/include $ sudo cp -P lib64/libcudnn* /usr/local/cuda-10.1/lib64 $ sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn* 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM