tensorflow gpu 安裝填坑記錄

環境

Anacoda
windows10 64操作系統專業版
1050TI顯卡

軟件包準備

  1. CUDA包
    cuda_10.0.130_411.31_win10.exe
    (鏈接:https://pan.baidu.com/s/1perhy_z4QbwDQPgam01Z1A 提取碼:f7ux
    自己去官方下載一定要注意版本,最新版本可能不適用)
  2. 顯卡驅動包
    417.71-desktop-win10-64bit-international-whql.exe(1050TI 驅動。你可以根據自己的顯卡下載驅動包,但是需要注意版本

注意一定要按照官網上的版本要求,否則會出現一些奇怪的錯誤。
實測在win10環境下不需要安裝後邊三個畫黑色×的包(應該是在linux環境下需要),只需要顯卡驅動CUDA
在這裏插入圖片描述
顯卡驅動和cuda安裝一路全選默認即可

tensorflow安裝

#創建name爲tensorflow的虛擬環境,其中python版本爲3.6
conda create -n tensorflow e python=3.6

#在當前CMD窗口環境切換到 tensorflow蓄力環境
activate tensorflow

#安裝tensorflow
pip install tensorflow-gpu

#測試
python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

如果報錯務必檢查所安裝軟件的版本是否符合官網要求。直接百度所拋出的錯誤,往往會走更多彎路。
官網參考:
https://tensorflow.google.cn/install

小彩蛋

jupyter 中使用 tensorflow的虛擬環境

#切換環境
activate tensorflow
#安裝ipykernel
pip install ipykernel
#將當前環境作爲一個新的jupyter的kernel
python -m ipykernel install --name tensorflow

完成之後打開jupyter notebook 就可以看到我們剛剛新建的kernel,使用這個kernel時,就是使用了tensorflow的虛擬環境
在這裏插入圖片描述

jupyter使用自動提示插件

#在base環境下操作
安裝jupyter_contrib_nbextensions
pip install jupyter_contrib_nbextensions
執行
jupyter contrib nbextension install --user

然後你的jupyter 就會對一個tab。這個裏邊有超級多好玩的功能,你可以去搜索更多有趣的玩法。本文中勾選的是自動提示的功能。
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章