前提条件:已经成功安装CUDA和cuDNN,以及Anconda软件。
1. 打开Anaconda Prompt
“开始” ——> Anaconda3(64-bit) ——> Anaconda Prompt (anaconda3)
2. 查看conda版本
conda --version
3. 查看现有虚拟环境
conda info --envs
4. 在命令窗口中输入命令创建环境
conda create -n xxx python=3.8
“xxx”代表所创建虚拟环境的名称
5. 激活所创建的虚拟环境
conda activate xxx
6. 安装TensorFlow
TensorFlow-gpu版本下载Links for tensorflow-gpu (tsinghua.edu.cn)
版本型号要和Anaconda保持一致。
下载好后,在当前环境下切换到下载目录,使用下面命令进行安装
pip install tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl
注:也可以选择在线安装pip install --ignore-installed --upgrade tensorflow-gpu
pip install --ignore-installed --upgrade tensorflow-gpu
7. 安装完成,验证版本
注意:tf.__version__前后是两个_组成。