安装cudnn
网上搜到了一些安装cudnn的步骤,都是这样的:1. 从官网下载cudnn2. 解压3. 将解压出来的lib64和include中所有的内容都拷贝到/usr/local/cuda的lib64和include中不行。看了一下解压出来的内容,lib64里面包含了两个软链接,直接cp过去软链接会失效。正确的做法是把lib64的路径添加到环境变量。正确的安装步骤1. 从官网下载cudnnNVIDIA cu
网上搜到了一些安装cudnn的步骤,都是这样的:
1. 从官网下载cudnn
2. 解压
3. 将解压出来的lib64和include中所有的内容都拷贝到/usr/local/cuda的lib64和include中
不行。
看了一下解压出来的内容,lib64里面包含了两个软链接,直接cp过去软链接会失效。
解决办法有两个:
1. 用 cp -d,可以复制链接关系
2. 把lib64的路径添加到环境变量
正确的安装步骤
1. 从官网下载cudnn
NVIDIA cuDNN | NVIDIA Developer
第一次下要注册一下
我的配置是tensorflow-gpu-1.15.0,cuda 10.0,下载对应版本的cudnn就好
查看合适的版本:
2. 解压
tar vxf cudnn-10.0-linux-x64-v7.6.0.64.tgz
3. 复制lib64下的内容
这一步有两个解决办法
1)cp -d
sudo cp -r -d [解压出来的lib64文件夹/*] usr/local/cuda/lib64/
2)修改环境变量
解压后会生成一个名叫cuda的文件夹
我把这个文件夹重命名成了 cudnn_7.6.0,然后把它移动到了/usr/local/
所以解压出来的lib64 的路径是
/usr/local/cudnn_7.6.0/lib64
把上面这个路径添加到LD_LIBRARY_PATH中
4. 复制include下的内容
把解压出来的include中的内容复制到 /usr/local/cuda/include中
sudo cp /usr/local/cudnn_7.6.0/include/* usr/local/cuda/include/
5. 重置读写权限
cudnn_7.6.0/include/ 里面只有一个 cudnn.h
sudo chmod a+r /usr/local/cuda/include/cudnn.h
为武汉地区的开发者提供学习、交流和合作的平台。社区聚集了众多技术爱好者和专业人士,涵盖了多个领域,包括人工智能、大数据、云计算、区块链等。社区定期举办技术分享、培训和活动,为开发者提供更多的学习和交流机会。
更多推荐
所有评论(0)