博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
CentOS 7安装tensorflow 1.6
阅读量:5843 次
发布时间:2019-06-18

本文共 1005 字,大约阅读时间需要 3 分钟。

1 pip3 install tensorflow-gpu==1.6

1.6 依赖于cuda 9.0 cuDNN 7,并且前提需要下载和安装nvidia驱动。

安装驱动

nvidia 驱动地址: 

1 vim /lib/modprobe.d/dist-blacklist.conf

添加:

1 blacklist nouveau2 options nouveau modeset=0

然后执行:

1 mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img2 dracut /boot/initramfs-$(uname -r).img $(uname -r)

重启,登录界面那里按ctrl+alt+f2,输入init 3。然后在命令行中执行那个run文件。

1 ./NVIDIA-Linux-x86_64-390.25.run

重启。

安装cuda

cuda 下载:

1 rpm -i cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm2 yum install cuda-libraries-9-03 yum install cuda-9-0.x86_64 --skip-broken4 ldconfig

安装cuDNN

参照官方说明:

cuDNN下载:

下载完拷到cuda9.0的目录下,再ldconfig。注意要下载7.0.4的。

1 cp include/* /usr/local/cuda-9.0/include/ 2 cp lib64/libcudnn* /usr/local/cuda-9.0/lib64/ 3 cd /usr/local/cuda-9.0/include/ 4 chmod a+r cudnn.h 5 cd /usr/local/cuda-9.0/lib64/ 6 chmod a+r libcudnn* 7 rm libcudnn.so 8 rm libcudnn.so.7 9 ln -s libcudnn.so.7.0.4 libcudnn.so.710 ln -s libcudnn.so.7 libcudnn.so11 ldconfig

 

转载于:https://www.cnblogs.com/linyx/p/8594040.html

你可能感兴趣的文章
将txt文件转化为json进行操作
查看>>
我的2014-相对奢侈的生活
查看>>
Java设计模式
查看>>
基本概念复习
查看>>
红黑树
查看>>
【数据库】
查看>>
WindowManager.LayoutParams 详解
查看>>
安卓中数据库的搭建与使用
查看>>
.NET 设计规范--.NET约定、惯用法与模式-2.框架设计基础
查看>>
sql 内联,左联,右联,全联
查看>>
C++关于字符串的处理
查看>>
Breaking parallel loops in .NET C# using the Stop method z
查看>>
[轉]redis;mongodb;memcache三者的性能比較
查看>>
让你的WPF程序在Win7下呈现Win8风格主题
查看>>
构建Docker Compose服务堆栈
查看>>
浮点数内存如何存储的
查看>>
JsonCpp 的使用
查看>>
问题账户需求分析
查看>>
hp 服务器通过串口重定向功能的使用
查看>>
此博客不再发表对自己私事的看法
查看>>