tensorflow-GPU安装文档
1.Anaconda安装
官网下载地址
https://www.anaconda.com/products/individual
安装流程
注意:此处要****勾选第一项Add Anaconda3 to my PATH environment variable****
2.安装tensorflow
1、打开Anaconda Prompt
2.在Anaconda Prompt中更换清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
即可添加 Anaconda Python 免费仓库。
3.新建一个TF2.1的环境 用python3.7的版本
conda create -n TF2.1 python=3.7
4.安装tensorflow
首先进入tensorflow环境
conda activate TF2.1
5.安装英伟达的SDK10.1版本
conda install cudatoolkit=10.1
出现以上界面一路y即可
同理,安装英伟达深度学习包软件
conda install cudnn=7.6
如果出错,说明电脑不支持,或者有可能是源没挂对,自行百度解决
3.安装tensorflow
pip install tensorflow==2.1
- 进入 python 验证是否安装成功
进入python
输入import tensorflow as tf 导入
然后输入 tf.version 查看版本
(TF2.1) C:\Users\tsl115>python
Python 3.7.11 (default, Jul 27 2021, 09:42:29) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
2022-02-28 20:39:57.511027: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
>>> tf__version
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
NameError: name 'tf__version' is not defined
>>> tf.__version__
'2.1.0'
>>>
如果显示和我差不多,说明安装成功
3.安装pycharm社区版
直接进官网下载社区版即可,具体百度就有,不会csdn上有很多教程
需要注意这一步,其他傻瓜式
进入pycharm进行配置测试
输入代码进行测试
import tensorflow as tf
tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()
print("tensorflow version:", tensorflow_version, "\tGPU available:", gpu_available)
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name="add")
print(result)