tensorflow gpu安装

发布时间：2020-12-14 01:08:28 所属栏目：Linux 来源：网络整理

导读：1.cuda 按官网命令 https://developer.nvidia.com/cuda-90-download-archive?target_os=Linuxtarget_arch=x86_64target_distro=CentOStarget_version=7target_type=rpmnetwork 更改/usr/local 软链接cuda指向新路径? 查看版本 cat /usr/local/cuda/version.t

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!