關(guān)于windows下Tensorflow和pytorch安裝教程

一、Tensorflow安裝

1、Tensorflow介紹

Tensorflow是廣泛使用的實現(xiàn)機器學習以及其它涉及大量數(shù)學運算的算法庫之一。Tensorflow由Google開發(fā)，是GitHub上最受歡迎的機器學習庫之一。Google幾乎在所有應用程序中都使用Tensorflow來實現(xiàn)機器學習。 例如，如果您使用到了Google照片或Google語音搜索，那么您就間接使用了Tensorflow模型。它們在大型Google硬件集群上工作，在感知任務方面功能強大。

2、Tensorflow安裝（cpu版本）

我的環(huán)境是Anaconda3.5.2m，安裝的是舊版本的tensorflow

pip install tensorflow==1.8.0

3、新版本安裝

pip install tensorflow
pip install -U --ignore-installed wrapt enum34 simplejson netaddr
pip install tensorflow

4、驗證代碼

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

5、警告Your CPU supports instructions that this TensorFlow binar......處理

#忽略警告處理方法
import os 
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 

6、GPU版本安裝（win10+1050ti)1、下載安裝CUDA（找到安裝包一直下一步）

2、安裝CUDNN(需要登錄才能下載)

解壓CUDNN安裝包

打開CUDA安裝位置

把CUDNN中的bin、lib、include目錄中的文件復制到CUDA對應的目錄中

3、安裝tensirflow-gup

pip install tensorflow-gpu==1.8.0

二、pytorch安裝

pytorch官網(wǎng)：https://pytorch.org/

1、pytorch介紹

pytorch是一個python優(yōu)先的深度學習框架，是一個和tensorflow，Caffe，MXnet一樣，非常底層的框架。

Torch 自稱為神經(jīng)網(wǎng)絡界的 Numpy, 因為他能將 torch 產(chǎn)生的 tensor 放在 GPU 中加速運算 (前提是你有合適的 GPU), 就像 Numpy 會把 array 放在 CPU 中加速運算. 所以神經(jīng)網(wǎng)絡的話, 當然是用 Torch 的 tensor 形式數(shù)據(jù)最好。 就像 Tensorflow 當中的 tensor 一樣。pytorch是一個動態(tài)的建圖的工具。不像Tensorflow那樣，先建圖，然后通過feed和run重復執(zhí)行建好的圖。相對來說，pytorch具有更好的靈活性。

2、安裝

打開 pytorch官網(wǎng) ，然后根據(jù)情況選擇安裝的版本。

我的python版本是3.6.5

pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

3、驗證代碼

import torch
print(torch.__version__)

總結(jié)

以上所述是小編給大家介紹關(guān)于windows下Tensorflow和pytorch安裝教程，希望對大家有所幫助！