win10+CUDA11.0+cuDNN8.0+anaconda3+python3.8+tensorflow_gpu 2.4.0+jupyter notebook 踩坑实录

原创已于 2023-06-03 13:00:18 修改 · 451 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#tensorflow #jupyter #python #cuda #配置环境

于 2023-06-03 12:59:14 首次发布

环境配置专栏收录该内容

1 篇文章

订阅专栏

本文详细记录了在Windows 10上安装CUDA 11.0、cuDNN 8.0、Anaconda3、Python 3.8以及TensorFlow-GPU 2.4.0的过程，并遇到的问题，如Python和iPython中导入TensorFlow失败（报错No module named 'gast'），以及在Jupyter Notebook中添加内核的步骤。此外，还介绍了如何删除和重新配置虚拟环境，以及在安装过程中对其他库如keras、sklearn和imblearn的依赖处理。

CUDA，cuDNN下载安装

参见 https://blog.csdn.net/Qi__Xi/article/details/127127629
注意【版本匹配】，我这里是CUDA11.0+cuDNN8.0+tensorflow_gpu 2.4.0
在这里插入图片描述

配置tensorflow-gpu

运行anaconda prompt （用管理员）

// 在anaconda下创建虚拟环境，命名为tf240
conda create -n tf240 python=3.8 
//激活tf2.4环境
conda activate tf240
//安装tensorflow-gpu 2.4.0版本
pip install tensorflow-gpu==2.4.0

至此，可以试试能不能成功

// 在虚拟环境 tf240 下，进入python
python 
//进入python环境
import tensorflow as tf
tf.test.is_gpu_available()
exit() # 退出，python环境也可使用ctrl+ZIP

在这里插入图片描述

进入ipython测试

// 在虚拟环境 tf240 下，进入ipython
ipython 
//进入ipython环境
import tensorflow as tf
tf.test.is_gpu_available()
exit()  #退出

在这里插入图片描述

踩坑开始

安装完，python环境不能import tensorflow，报错

// 在虚拟环境 tf240 下
pip install protobuf==3.19.0

安装完，ipython环境不能import tensorflow，报错 No module named ‘gast’

// 在虚拟环境 tf240 下
conda install ipython

在jupyter notebook调用，增加内核

在【虚拟环境 tf240】下，# conda activate tf240 #进入该环境

// 在虚拟环境 tf240 下，一定！！！
pip install ipykernel ipython -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
python -m ipykernel install --user  --name tf240 --display-name "tf240"

如果上述解决方法不行，想要删除重来

删除虚拟环境

// 删除虚拟环境，在anaconda prompt默认环境下
// 查看虚拟环境名字
conda env list 
// conda remove -n  需要删除的环境名 --all
conda remove -n tf240 --all

删除 jupyter 内核

// 删除 jupyter 内核，在anaconda prompt默认环境下
// 查看内核名字
jupyter kernelspec list
// jupyter kernelspec remove 内核名字
jupyter kernelspec remove tf240

调用库

以下库的安装都是在【虚拟环境 tf240】下，# conda activate tf240 #进入该环境
在import 其它库时，如果调用错误，试试更改numpy版本

// numpy版本改为numpy-1.20.3
pip uninstall numpy
pip install numpy==1.20.3

keras 需要 from tensorflow import keras
sklearn 安装是 pip install scikit-learn
imblearn 安装是 pip install imbalanced-learn