Win10下安裝TensorFlow(GPU版本)

2021-08-01 08:01:44 字數 1601 閱讀 1605

之前在用的 tensorflow 是 cpu 版本的,現在感覺雖然自己電腦的 gpu 沒有多強,但還是很有用的,因此把 tensorflow 換成了 gpu 版本的。安裝過程比較順利,但也有一些值得注意的地方,與大家分享下。

要注意 tensorflow 需要 python3.5 以上的,別下錯了

有了 anaconda ,安裝 tensorflow 就很簡單了

只需要在 cmd 輸入以下命令就可以了。

如上圖所示,左邊為 cudnn 解壓得到的檔案,右邊為 cuda 的安裝位置。

安裝 cuda 與安裝 tensorflow 沒有順序關係,先安裝誰都可以。

在 python 中執行以下測試**:

import tensorflow as tf

#creates a graph.

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

#creates a session with log_device_placement set to true.

sess = tf.session(config=tf.configproto(log_device_placement=true))

#runs the op.

print(sess.run(c))

如果你得到類似以下輸出,則說明你已經成功的安裝了 gpu 版本的 tensorflow 了

/job:localhost/replica:0/task:0/gpu:0 -> device: 0, name: tesla k40c, pci bus

id: 0000:05:00.0

b: /job:localhost/replica:0/task:0/gpu:0

a: /job:localhost/replica:0/task:0/gpu:0

matmul: /job:localhost/replica:0/task:0/gpu:0

[[ 22. 28.]

[ 49. 64.]]

此外 如果你是在 anaconda 的 notebook中執行測試**的,可能輸出的只有計算矩陣的結果:

在win10 安裝cuda和tensorflow

因為自己的筆記本也有gpu,加上tensorflow 增加了windows的支援,所以打算裝乙個tf。打算裝python版的tf,首先當然是裝python了。python 3.6 1 anaconda 4.4 0 64 bit default,may 11 2017,13 25 24 msc v.1...

Win10下安裝tensorflow gpu版)

系統 win10系統 python 3.5 gpu geforce gtx1660ti cuda cuda 9.0.176 win10 cudnn cudnn 9.0 windows10 x64 v7 1.安裝anaconda 2.安裝vs2017 vs2015也可以 3.安裝cuda9.0 cud...

不是變數 win10 在win10下安裝manim

manim是製作簡單動畫的乙個python庫,具體介紹請參考 3b1b manim github.com 例如 無論是windows或者是其它作業系統使用者的安裝方式,在鏈結中有詳細說明。下面詳細說明一下安裝步驟,重點放在win10使用者可能遇到的問題上。anaconda,texlive,ffmpe...