前言

學習隨筆，權作記錄。

一、tensorflow-cpu指定版本的卸載

>> pip uninstall tensorflow==版本號
例如：
>> pip uninstall tensorflow==2.1.0

二、tensorflow-gpu指定版本的卸載

>> pip uninstall tensorflow==版本號
例如：
>> pip uninstall tensorflow-gpu==2.1.0

三、tensorflow-cpu指定版本的安裝

1、創(chuàng)建虛擬環(huán)境

打開Anaconda prompt（Anaconda3）

檢查當前環(huán)境下python的版本

>> python --version

根據python的版本，創(chuàng)建虛擬環(huán)境pachong

>> conda create -n pachong python==3.7.4

按“y”，再按回車即可。

輸入conda info -e，查看全部的環(huán)境，可以看到pachong環(huán)境已經創(chuàng)建完成。

>> conda info -e

2、激活虛擬環(huán)境

在base環(huán)境下輸入conda activate pachong，即可進入pachong環(huán)境

>> conda activate pachong

3、安裝指定版本的tensorflow

講鏡像源切換為清華大學的鏡像

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/
# 設置搜索時顯示通道地址
conda config --set show_channel_urls yes

再輸入，等待安裝即可。

>> conda install tensorflow=版本號
例如：
>> conda install tensorflow=1.15.0

四、tensorflow-gpu指定版本的安裝

首先，根據python版本，確定tensorflow-gpu，cuda和cudnn的版本。

下文以安裝tensorflow_gpu-2.1.0為例。

安裝CUDA

在我們的具體使用中，其實真正需要的并不是整個CUDA，而是cudatoolkit，所以我們在這里直接安裝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，即可安裝。

測試tensorflow安裝成功

進入虛擬環(huán)境tensorflow

在虛擬環(huán)境下，打開python

在python下輸入import tensorflow as tf

在輸入tf.test.is_gpu_available()，即可。若最后為True表示GPU可用，證明已經成功安裝Tensorflow的GPU版本。

可能遇到的問題

如果安裝完成，測試的時候出現如下問題：

解決方案：根據描述的情況將protobuf的版本降到3.19.0以下即可。
具體命令如下：

pip install protobuf==3.19.0

總結

以上就是全部內容，除了換源之外還有類似于直接在命令中加鏡像源的下載方式，以及直接下載whl鏡像源文件，安裝的方式。有興趣的可以自行搜索。文章來源地址http://www.zghlxwxcb.cn/news/detail-417695.html

到了這里，關于【1】如何安裝和卸載tensorflow-CPU和GPU各版本-簡單清晰版的文章就介紹完了。如果您還想了解更多內容，請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章，希望大家以后多多支持TOY模板網！