首页 > 其他 > 详细

tensorflow gpu安装

时间:2019-05-21 20:28:30      阅读:103      评论:0      收藏:0      [点我收藏+]

1.cuda

按官网命令

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=rpmnetwork

更改/usr/local 软链接cuda指向新路径 

查看版本 cat /usr/local/cuda/version.txt 

2.cudnn

下载 https://developer.nvidia.com/cudnn

解压

$ cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz

$ tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装

https://blog.csdn.net/yaoqiwaimai/article/details/78594154

查看 CUDNN 版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

重装cuda后nvidia-smi报错

reboot

 

tensorflow gpu安装

原文:https://www.cnblogs.com/iwanna/p/10901991.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!