Tensorflow和pytorch安裝（windows安裝）

1、 Tensorflow安裝

一、 Tensorflow介紹

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

二、Tensorflow安裝（cpu版本）

個人環境是Anaconda3.5.2m，安裝的是舊版本的tensorflow

pip install tensorflow==1.8.0

三、新版本安裝

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

四、驗證代碼

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

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

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

 

 

六、GPU版本安裝（win10+1050ti)

一、下載安裝CUDA（找到安裝包一直下一步）

 

 

 

 

二、安裝CUDNN(須要登陸才能下載)

 

解壓CUDNN安裝包

 

打開CUDA安裝位置

 

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

 

三、安裝tensirflow-gup

pip install tensorflow-gpu==1.8.0

 

 

2、pytorch安裝

pytorch官網：https://pytorch.org/

一、pytorch介紹

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

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

二、安裝

打開 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

 

三、驗證代碼

import torch print(torch.__version__)