Tensorflow和pytorch安裝（windows安裝）

一、 Tensorflow安裝

1、 Tensorflow介紹

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

2、Tensorflow安裝（cpu版本）

我的環境是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官網：https://pytorch.org/

1、pytorch介紹

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

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

2、安裝

打開 pytorch官網 ，然后根據情況選擇安裝的版本。

 

 

 我的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__)