Tensorflow环境安装

最新推荐文章于 2024-07-09 17:03:37 发布

林老头、

最新推荐文章于 2024-07-09 17:03:37 发布

阅读量235

点赞数 1

分类专栏： DeepLearning 文章标签： cuda tensorflow

本文链接：https://blog.csdn.net/lin455989875/article/details/102324863

版权

DeepLearning 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文档详细介绍了在Windows和Linux系统中安装Tensorflow的步骤，包括安装Anaconda、CUDA、cuDNN，配置PATH环境变量，以及测试CUDA和Tensorflow的正确安装。在Windows环境下，特别强调了CUDA安装时的注意事项；在Linux环境下，提供了CUDA和cuDNN的安装及环境变量配置方法。

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

Tensorflow环境安装

Windows安装

一、安装Anaconda

点击这里下载Anaconda进行安装，跟普通软件一下双击打开安装即可。
安装时候勾选该项：

在这里插入图片描述

验证：在cmd窗口输入conda list

二、安装CUDA

点击这里下载CUDA10版本，下载后跟普通软件一样双击打开安装。
安装时候选择自定义安装避免报错：

在这里插入图片描述

去掉如下图的选项：

在这里插入图片描述

对于显卡驱动，win系统会安装显卡驱动，而CUDA也带有显卡驱动，此时需要保证的是：如果电脑安装的显卡驱动版本要高于CUDA自带的显卡驱动，则不需要选中下面选项：

在这里插入图片描述

验证：查看是否存在该文件

在这里插入图片描述

三、安装cuDNN

点击这里下载跟CUDA同版本的cuDNN
cuDNN不是应用软件，直接解压复制到CUDA安装文件夹即可。
验证：该文件目录下存在该文件：

在这里插入图片描述

四、PATH配置

在环境变量中做如下改变：

在这里插入图片描述

确认：

在这里插入图片描述

五、测试CUDA

在cmd中执行：nvcc -v

在这里插入图片描述

六、Tensorflow安装

目前Tensorflow提供1.x和2.0版本：

在这里插入图片描述

去py文件中执行：

import tensorflow as tf

# 查看tf的版本号
print(tf.__version__)
# 测试gpu版本是否可用
print(tf.test.is_gpu_available())
#输出True说明gpu正常

Linux安装

视频教程：点击查看

一、安装Anaconda

点击这里下载Anaconda进行安装

二、安装CUDA

点击这里下载Ubuntu18.04的CUDA10.0版本。

安装方式：

`sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb`
`sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`