Windows10 安裝tensorflow-gpu記錄

1 配置

系統(tǒng):Windows10 專業(yè)版

CPU:i5-4210m

顯卡:GeForce 940m

2 準(zhǔn)備

? ? 1.Python :python 3.6.5 (tensorflow2.0和最新的2.1都只支持到Python 3.7)

? ? 2.VS2017?:官網(wǎng)下載了后安裝時(shí)沒有點(diǎn)勾所有功能,只安裝了200M(感覺可以不裝。。),?還有Microsoft Visual C++ 2015 Redistributable Update

? ? 3.Cuda:下載官網(wǎng)Cuda10.0.1 和與其配對的cuDNN7.6.5(下載下來的cuDNN壓縮包解壓后,分別把解壓文件夾中bin,include和lib/x64中的文件復(fù)制到Cuda的安裝路徑中的bin,include和lib/x64 下)

3 安裝

? ? 1.pip安裝

在cmd中輸入 :

pip install pip -U? # 升級 pip 到最新的版本 (>=10.0.0) 后進(jìn)行配置

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple #換清華下載源更快

pip install tensorflow-gpu==2.0.0a0? #安裝gpu版2.0.0

(之前直接 pip install tensorflow 安裝了cpu版2.1.0的,又 pip uninstall tensorflow? 卸載,又

pip install tensorflow-gpu 安裝后import出錯(cuò):“找不到指定模塊”,換成2.0.0a0就可以import tnesorflow 了。)

? ??2.查看gpu是否可用

在python中運(yùn)行:

import tensorflow as tf

tf.test.is_gpu_available()

.結(jié)果為True 說明是gpu可用。

(因?yàn)橹鞍惭b了又卸載tensorflow ,最后安裝了gpu版的tensorflow,gpu也不是可用,可能是沒有卸載干凈,cmd中輸入 pip list ,發(fā)現(xiàn)其中還有tensorflow-estimator? 2.1.0 和?tensorflow-gpu-estimator? 2.1.0, 將兩個(gè)都卸載了

pip uninstall tensorflow-estimator

pip uninstall tensorflow-gpu-estimator

結(jié)果就為True了。)

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容