安装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
查看端口