在anaconda中安裝tensorflow-GPU版本

第一步：去anaconda官網(wǎng)下載，我下載的是anaconda (python3.7版本）

anaconda下載鏈接：點擊link
進去之后網(wǎng)頁最下面有要下載的版本
安裝的具體步驟：點擊
link

第二步：安裝tensorflow的具體步驟

1、 anaconda安裝之后進入Anaconda Prompt(anaconda)

2、進去之后創(chuàng)建一個虛擬環(huán)境
代碼如下
conda create -n TF2.1 python=3.7
輸入之后點擊回車


用coda activate TF2.1
激活之后不要急著安裝，因為下載會很慢，安裝中科大鏡像：具體步驟點擊link，安裝成功之后再進行以下安裝操作
激活這個虛擬環(huán)境，進入虛擬環(huán)境之后安裝
conda install cudatoolkit=10.1
安裝過程中出現(xiàn)Proceed（[y]/n)?都選y

安裝成功之后：安裝下面這個，步驟同上
conda install cudnn=7.6
成功之后安裝
pip install tensorflow==2.1(不用指定GPU版本，默認CPU版本，GPU版本都會下載，GPU不能用的時候轉(zhuǎn)為使用CPU）
安裝之后測試一下，還是當前頁面
輸入

python回車
import tensorflow as tf 回車
tf.__version__回車

安裝GPU版本需要Microsoft Visual C++ 2015-2019 Redistributabled的支持，木有安裝的點擊link自行安裝,里面有鏈接
link

下載完之后如何在anaconda中使用剛才創(chuàng)建的虛擬環(huán)境

進入anaconda Prompt，激活剛才的虛擬環(huán)境
conda activate TF2.1回車，進入環(huán)境之后輸入conda install nb_conda
回車
出現(xiàn)Proceed（[y]/n)?都選y

安裝之后，會出現(xiàn)jupyter notebook(TF2.1),點擊進去

點擊new，選擇剛才創(chuàng)建的虛擬環(huán)境Python [conda env:TF2.1]

不需要時可以刪除創(chuàng)建的虛擬環(huán)境

conda remove -n 虛擬環(huán)境名 --all

總結(jié)

以上是生活随笔為你收集整理的在anaconda中安装tensorflow-GPU版本的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。