tensorflow-gpu配置

1.安装Anaconda

下载地址：Anaconda | The Operating System for AI

2.查询tensorflow-gpu不同版本所对应的python、cuDNN和CUDA的版本

官网：在 Windows 环境中从源代码构建  |  TensorFlow (google.cn)

 3.使用conda安装相应的库

?

1 2 3 4 5 6 7 8 9

# tensorflow_gpu-2.3.0 conda init conda create --name tf conda activate tf conda install python=3.6 conda install cudatoolkit=10.1 conda install cudnn=7.6 pip install tensorflow_gpu==2.3.0 pip install keras==2.3.0

4.jupyter notebook配置

?

1 2 3 4

# jupyter notebook 配置 conda install ipykernel conda install nb_conda_kernels pip install jupyter notebook

随后在jupyter notebook中选择相应的内核即可：