在centos7 上完成tensorflow-gpu + anaconda + python3.6 + CUDA + CUDNN 部署

最新推荐文章于 2024-07-21 10:39:03 发布

luffysup

最新推荐文章于 2024-07-21 10:39:03 发布

阅读量926

点赞数

本文链接：https://blog.csdn.net/u014636669/article/details/95628259

版权

本文档详细介绍了如何在CentOS7系统中安装TensorFlow-GPU，首先从anaconda3的安装开始，通过创建Python3.6的虚拟环境来满足TensorFlow-GPU的需求。接着，逐步讲解了CUDA和CUDNN的安装过程，最后使用pip安装了特定版本的tensorflow-gpu以避免依赖冲突。完成这些步骤后，一个基本的深度学习开发环境就搭建完毕。

摘要由CSDN通过智能技术生成

step1. 安装anaconda

我选择的版本是anaconda3，自带的python版本为3.7

去官网下载对应的anaconda版本，这一步使用清华大学的anaconda源，参考地址如下

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

step2.配置python3.6的环境(因为tensorflow-gpu目前支持的最高版本是python3.6)

使用 conda create -n your_env_name python=3.6， anaconda 命令创建python版本为X.X、名字为your_env_name的虚拟环境。your_env_name文件可以在Anaconda安装目录envs文件下找到。

然后激活该环境

source activate your_env_name(虚拟环境名称)

step3.安装cuda

首先要有安装GPU的环境，然后就是安装官网下载的cuda9.0

   sudo yum install gcc    
   sudo yum install gcc-c++ 
   sudo yum install kernel-devel  
   sudo rpm -i cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm
   sudo yum clean all
   sudo yum install cuda

step4.安装cudnn

在官网下载与cuda对应的版本，然后安装，下载官网上版本后，会发现后缀名不对，需要修改一下