安装tensorflow-gpu


安装gpu驱动

此处生省略.....

对应版本

查看对应版本关系,这一步很重要,不然后续会报错

cpu版本

 gpu版本

安装Bazel

参考官网:https://docs.bazel.build/versions/master/install-redhat.html#installing-menu

cd /etc/yum.repos.d/
wget https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo
yum install bazel

安装CUDA

首先选择对应的版本:https://developer.nvidia.com/cuda-toolkit-archive

wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-rhel7-10-0-local-10.0.130-410.48-1.0-1.x86_64
sudo rpm -i cuda-repo-rhel7-10-0-local-10.0.130-410.48-1.0-1.x86_64.rpm
sudo yum clean all
sudo yum install cuda

安装cuDNN

cuDNN下载需要登录,可以自行注册,查看官网(https://developer.nvidia.com/rdp/cudnn-archive),
获取对应文件,下载到本地,通过传输工具再传到Centos系统中。

tar -xzvf cudnn-10.0-linux-x64-v7.4.2.24.tgz
cp cuda/include/cudnn.h /usr/local/cuda/include/
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 添加环境变量

vim /etc/profile

# 添加环境变量
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda

安装tensorflow-gpu

yum install python-devel python3-devel python3-pip
pip3 install -U pip
pip3 install tensorflow-gpu==1.13.1

验证

(gpu) [root@t8t ~]# python
Python 3.6.8 |Anaconda, Inc.| (default, Dec 30 2018, 01:22:34) 
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.test.is_built_with_cuda()
True

nginx + uwsgi + flask部署到centos7

编辑uwsgi.ini文件

[uwsgi]
http = :8080
chdir = /root/final_tower
processes = 8
threads = 30
pidfile = uwsgi.pid
master = true
wsgi-file = /root/final_tower/final_tower.py
callable = app
daemonize = uwsgi.log
#anaconda3 = /root/anaconda3/envs/tensorflow/bin/python3
vacuum = true

启动uwsgi

uwsgi uwsgi.ini

部署nginx,nginx.conf配置如下,可以做适当修改

upstream xxx{
        server 127.0.0.1:8080;
        keepalive 32;
        keepalive_timeout 80s;
    }

    server {
        listen       80;
        listen       [::]:80;
        server_name  _;
        root         /usr/share/nginx/html;
        large_client_header_buffers 4 16k;
        client_max_body_size 300m;
        client_body_buffer_size 128k;
        proxy_connect_timeout 600;
        proxy_read_timeout 600;
        proxy_send_timeout 600;
        proxy_buffer_size 64k;
        proxy_buffers 4 32k;
        proxy_busy_buffers_size 64k;
        proxy_temp_file_write_size 64k;
        # Load configuration files for the default server block.
        include /etc/nginx/default.d/*.conf;
        location / {
            proxy_pass http://xxx;
            proxy_connect_timeout 600;
            proxy_send_timeout 600;
            proxy_read_timeout 600;
        }

        error_page 404 /404.html;
        location = /404.html {
        }

        error_page 500 502 503 504 /50x.html;
        location = /50x.html {
        }
    }

启动nginx

systemctl restart nginx

查看端口

相关