TensorFlow笔记（1）环境搭建： win10+cuda10.0+cudnn10.0+anaconda+ tensorflow-gpu+vscode+python

最新推荐文章于 2023-10-15 19:49:01 发布

libratianc

最新推荐文章于 2023-10-15 19:49:01 发布

阅读量499

点赞数

分类专栏： TensorFlow-gpu 文章标签： TensorFlow

本文链接：https://blog.csdn.net/fu373323265/article/details/97620564

版权

1 篇文章 0 订阅

订阅专栏

第一步：先更新英伟达驱动到最新版

第二步：安装cuda10.0驱动，这里注意为避免TensorFlow兼容性问题 cuda安装10版本就足够学习和使用了，

注意安装位置就选择默认的位置，如果更换位置，可能给以后的操作带来不便

安装cuda时，选择自定义安装安装内容全部打钩安装，其他全部默认设置就行

第三步：安装cudnn10.0，这里为了配套，安装10.0就行

cudnn10.0其实就是压缩文件，解压后得到三个文件夹，把这个三个文件夹放入cuda对应的文件夹中就行

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64

再将上面三个路径添加到环境变量中的path中。

一般是上面三个位置，环境变量添加完是这个样子

第四步：安装anaconda3，这个没有版本限制，推荐最新版

第五步：在anconda3中创建新的工作环境，python版本选择3.7

第六步：下载TensorFlow-gpu，国内环境下注意使用国内镜像源:

第七步：测试TensorFlow-gpu是否安装成功

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

如果没有异常报错就算安装成功了，安装成功的话会显示

本笔记中用到的软件下面地址中可以下载到

链接: https://pan.baidu.com/s/1rLpjzpO5je-6CQ6-1Ksr1A 提取码: vnm2 复制这段内容后打开百度网盘手机App，操作更方便哦

关注