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中选择相应的内核即可: