在anaconda中安裝tensorflow-GPU版本

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

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

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

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

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


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

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

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

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

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

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

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

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

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

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

總結(jié)

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

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