tensorflow-gpu配置

1.安装Anaconda

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

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

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

 3.使用conda安装相应的库

# 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配置

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

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

posted @ 2024-07-01 17:29  peach1  阅读(11)  评论(0编辑  收藏  举报