[TensorFlow]Windows下安装并运行Hello World

最新推荐文章于 2024-06-28 21:47:40 发布

weixin_30407099

最新推荐文章于 2024-06-28 21:47:40 发布

阅读量332

点赞数

原文链接：http://www.cnblogs.com/hcbin/p/11324712.html

版权

参考网址：https://www.tensorflow.org/install/pip (或要VPN)

建议电脑是独显N卡机，安装前先升级驱动，减少不必要错误

1.下载Microsoft Visual C++ 2015 Redistributable Update 3（已安装更高级版或安装VS2015或VS2017可以跳过）

https://www.microsoft.com/zh-CN/download/details.aspx?id=53587

2.启用长路径。
https://superuser.com/a/1119980

3. 安装Python3.7.4，在安装界面上勾选添加环境变量复选框

https://www.python.org/downloads/

修改pip源（永久）

新建C:\Users\Administrator\pip\pip.ini，添加代码：

[global]
timeout=6000
index-url=https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host=pypi.tuna.tsinghua.edu.cn

批量更新包（可选）

参考：http://blog.sciencenet.cn/home.php?mod=space&uid=478347&do=blog&id=1115281

import pip
# pip V10.0.0以上版本需要导入下面的包
from pip._internal.utils.misc import get_installed_distributions
from subprocess import call

for dist in get_installed_distributions():
    call('python -m pip install --upgrade --user ' + dist.project_name, shell=True)

PS：numpy不要装得太新（v1.17.0+），不然有很多警告

python -m pip install numpy==1.16.4

4.cuda 10.0和cudnn 7.6.1

https://developer.nvidia.com/cuda-downloads

https://developer.nvidia.com/rdp/cudnn-archive （注意对应cuda版本）

cudnn安装方法参考：https://jingyan.baidu.com/article/39810a236d660bb636fda6d4.html（简单来说就是解压cudnn，将文件粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0）

添加环境变量到PATH

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

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

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

PS：我的电脑不支持10.1，当运行hello world时就会报错

ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: https://developer.nvidia.com/cuda-90-download-archive

5. 安装tensorflow-gpu 1.14.0

pip install tensorflow-gpu==1.14.0

6.测试

import tensorflow as tf

h = tf.constant("Hello")
w = tf.constant(" World!")
hw = h + w

# 警告信息提示要将 tf.Session() 替换为 tf.compat.v1.Session()
with tf.compat.v1.Session() as sess:  
    ans = sess.run(hw)

print(ans)

屏蔽GPU输出信息可以添加上这句：

参考：https://blog.csdn.net/dcrmg/article/details/80029741

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'

TF_CPP_MIN_LOG_LEVEL
0 ：输出所有信息（默认值）
1 ：屏蔽通知信息
2 ：屏蔽通知信息和警告信息
3 ：屏蔽通知信息、警告信息和报错信息

关于Anaconda安装

只需要将上面的python安装替换为Anaconda（好像会自带python，没测试）

1.安装Anaconda 2019.07（注意对应Python版本）

https://www.anaconda.com/distribution/

# 指定升级某一个库，例如 pandas
conda update pandas

# 升级所有可升级的库
conda update --all

2.安装OpenSSL v1.1.1c，不然Conda安装时可能会出下面的错误

http://slproweb.com/products/Win32OpenSSL.html

3.打开CMD，添加Conda源（可以不要https）

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

PS：可以直接打开C:\Users\Administrator\.condarc编辑：

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
show_channel_urls: true

7. 安装tensorflow-gpu 1.14.0