Win10下用Anaconda安装TensorFlow-gpu版本

一、本地安装

如果直接在本地环境安装（不创建虚拟环境:参考下文虚拟环境）。直接这样

 

 

conda install -c conda-forge cudatoolkit=11.2.0 cudnn=8.1.0
conda install tensorflow-gpu==2.10

（最好改一下conda下载源）

中科大anaconda源参考链接中科大anaconda源

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

二、虚拟环境安装

建议在Anaconda Prompt (anaconda3)中运行

1、查看所有虚拟环境

conda info --envs

2、安装虚拟环境

建议3.9.0之前版本，ipykernel只支持3.9.0之前版本（ipykernel是支持在jupyter notebook打开虚拟环境的库）

如果直接在本地，python版本不能超过3.10 

 

conda create --name tf-gpu python==3.8.0

--name 后面的是虚拟环境名字，也就是自定义环境名称，譬如：tf-gpu

python==是指定安装的版本号。这里安装的版本是3.8.0

3、进入虚环境tf-gpu
 

activate tf-gpu

# 退出虚环境   deactivate或者conda deactivate

# 删除虚环境   conda remove --name tf-gpu --all

4、查看本机支持哪些tensorflow版本

（带gpu是gpu版本，默认cpu版本）

GPU版本

conda search tensorflow-gpu

或者
CPU版本

conda search tensorflow

安装tensorflow（注：安装GPU版本系统会自动根据机器环境安装相应的CUDA和cuDNN）

如上图，安装对应的tensorflow-gpu版本需要 安装对应的cuDNN和CUDA。

tensorflow-gpu2.11之后就不再支持windows安装。

所以最新只能安装到tensorflow-gpu==2.10

版本和cuda、cudnn必须匹配（不然各种bug）

个人建议：先去CUDA官网下载cuda
再去cuDNN官网下载cuda对应的cuDNN

！！！     先cuda
接着cuDNN  
  
          最后再conda install tensorflow-gpu

# CUDA和cuDNN也可以不下载，

但是conda install tensorflow-gpu的时候会下载CUDA和cuDNN这两个的部分插件，外网下载速度很慢。不如提前装好CUDA和cuDNN

英伟达驱动下载网站：CUDA

安装好CUDA之后，再安装cuDNN

查询下载cuda对应版本的cuDNN

英伟达驱动下载： cuDNN

下载cuDNN需要登陆权限，如果觉得注册账号复杂的话，可以从在云盘上下载

cuDNN_8.8.0 （CUDA 12.X）

        迅雷云盘下载

        百度网盘下载

cuDNN_8.7.0（CUDA 11.X）

        迅雷云盘下载

        百度网盘下载

下载好cuDNN将其解压，把cuDNN中的文件复制到对应的CUDA文件中，

5、安装TensorFlow-gpu2.6版本

也可以不带==2.6.0（默认安装最新版本，等等是指定版本）

conda install tensorflow-gpu==2.6.0

 安装TensorFlow2.6版本

也可以不带==2.6.0（默认安装最新版本，等等是指定版本）

conda install tensorflow==2.6.0

6、（1）在原始环境中安装插件和ipykernel（在jupyter中可以选择环境）

conda install nb_conda
conda install ipykernel

（2）虚环境中安装ipykernel
 

activate ➕自定义环境名称  # 进入指定环境中

activate tf-gpu         
conda install ipykernel

注：一般会使用Matplotlib包，虚环境中没有这个包，需要手动安装！！！
# 因为安装ipykerner时conda会自动升级，再使用conda安装其它包时可能会报错，建议使用pip命令安装

pip install matplotlib

7、检验TensorFlow2.0是否安装成功

确认在自定义环境名中  # activate tf-gpu

进入python环境

python

 导入TensorFlow

import tensorflow as tf

查看tensorflow版本

print(tf.__version__)

查看GPU

print('GPU', tf.test.is_gpu_available())

# 最好不要安装最新的tensorflow，本地不支持(gpu)
# 经测试，tensorflow-gpu的版本是2.10，python版本是3.9，可使用

# 安装代码如下
# 1安装python虚拟环境
conda create --name tf-gpu python==3.9
# 2进入虚拟环境
activate tf-gpu
# 查看可安装版本
conda search tensorflow-gpu

#安装tensorflow-gpu2.10版本
pip install tensorflow-gpu==2.10.0

使用 pip 安装 TensorFlow

https://tensorflow.google.cn/install/pip?hl=zh-cn#package-location