前言
學(xué)習(xí)隨筆，權(quán)作記錄。
一、tensorflow-cpu指定版本的卸載
>> pip uninstall tensorflow==版本號(hào)
例如：
>> pip uninstall tensorflow==2.1.0
二、tensorflow-gpu指定版本的卸載
>> pip uninstall tensorflow==版本號(hào)
例如：
>> pip uninstall tensorflow-gpu==2.1.0
三、tensorflow-cpu指定版本的安裝
1、創(chuàng)建虛擬環(huán)境
打開Anaconda prompt（Anaconda3）
檢查當(dāng)前環(huán)境下python的版本
>> python --version
根據(jù)python的版本，創(chuàng)建虛擬環(huán)境pachong
>> conda create -n pachong python==3.7.4
按“y”，再按回車即可。
輸入conda info -e，查看全部的環(huán)境，可以看到pachong環(huán)境已經(jīng)創(chuàng)建完成。
>> conda info -e
2、激活虛擬環(huán)境
在base環(huán)境下輸入conda activate pachong，即可進(jìn)入pachong環(huán)境
>> conda activate pachong
3、安裝指定版本的tensorflow
講鏡像源切換為清華大學(xué)的鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/
# 設(shè)置搜索時(shí)顯示通道地址
conda config --set show_channel_urls yes
再輸入，等待安裝即可。
>> conda install tensorflow=版本號(hào)
例如：
>> conda install tensorflow=1.15.0
四、tensorflow-gpu指定版本的安裝
首先，根據(jù)python版本，確定tensorflow-gpu，cuda和cudnn的版本。
下文以安裝tensorflow_gpu-2.1.0為例。
安裝CUDA
在我們的具體使用中，其實(shí)真正需要的并不是整個(gè)CUDA，而是cudatoolkit，所以我們?cè)谶@里直接安裝cudatoolkit
激活虛擬環(huán)境conda activate pachong
輸入conda conda install cudatoolkit=10.1，安裝10.1版本的cudatoolkit。
大概是300MB左右。
安裝cuDNN
再pachong環(huán)境下，輸入conda install cudnn=7.6，安裝cudnn的7.6版本。
大概200MB左右。
安裝tensorflow_gpu-2.1.0
安裝tensorflow使用pip安裝，在虛擬環(huán)境下輸入pip install tensorflow-gpu==2.1.0，即可安裝。
測(cè)試tensorflow安裝成功
進(jìn)入虛擬環(huán)境tensorflow
在虛擬環(huán)境下，打開python
在python下輸入import tensorflow as tf
在輸入tf.test.is_gpu_available()，即可。若最后為True表示GPU可用，證明已經(jīng)成功安裝Tensorflow的GPU版本。
可能遇到的問題
如果安裝完成，測(cè)試的時(shí)候出現(xiàn)如下問題：
解決方案：根據(jù)描述的情況將protobuf的版本降到3.19.0以下即可。
具體命令如下：文章來源：http://www.zghlxwxcb.cn/news/detail-402176.html
pip install protobuf==3.19.0
總結(jié)
以上就是全部?jī)?nèi)容，除了換源之外還有類似于直接在命令中加鏡像源的下載方式，以及直接下載whl鏡像源文件，安裝的方式。有興趣的可以自行搜索。文章來源地址http://www.zghlxwxcb.cn/news/detail-402176.html
到了這里，關(guān)于【1】如何安裝和卸載tensorflow-CPU和GPU各版本-可能不快的但是簡(jiǎn)單版的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
![Linux和Windows系統(tǒng)下:安裝Anaconda、Paddle、tensorflow、pytorch，GPU[cuda、cudnn]、CPU安裝教學(xué),以及查看CPU、GPU內(nèi)存使用情況](https://imgs.yssmx.com/Uploads/2024/02/762289-1.png)
