python3.6.4+anaconda5.1.0+cuda9.0.176+cudnn_v7.4.1.5+tensorflow安装心得
前言
最近做本科毕业设计,要用到图卷积神经网络,开始入坑深度学习。在配置深度学习环境时踩了很多坑,为防止自己和大家继续踩雷,在此记录。我遇到的问题有:
- python版本太高,配置不兼容
- cuda版本和电脑显卡不匹配
- tensorflow版本不匹配
总之,都是粗心大意,没有注意版本匹配问题。在浏览了众多帖子后,最终采用最稳妥的方式,python3.6.4+anaconda5.1.0+cuda9.0.176+cudnn_v7.4.1.5+tensorflow。
文中,附了其他博主的安装心得,侵删。
python3.6.4
在python官网下载所需的版本
anaconda5.1.0
- anaconda下载地址
注意:官网上是最新版本,并不是最新就是好的,而是要版本兼容,上述地址是清华镜像地址,下载快,各版本都有。 - anaconda与python版本对应说明link
- anaconda简介link
- anaconda prompt命令行操作指令介绍
cuda9.0
cuda9.0下载地址
cuda10.0下载地址
cuda版本选择说明link*
tensorfolow
安装参考1
tensorflow版本更新
在pycharm上搭建tensorflow
测试
利用anaconda prompt安装过程
注意:conda install与pip install区别link
强烈建议用anaconda配置,这样方便用pycharm导入已安装的第三方库
参考文章1
参考文章2
⭐参考文章3
- 加载源以便快速安装
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
- 创建虚拟环境
conda create -n tensorflow1_12 python=3.6.4
- 激活虚拟环境
activate tensorflow1_12
- 寻找要安装的tensorflow版本并安装
conda search tensorflow-gpu
conda install tensorflow-gpu=1.12.0
一些问题的解决方法(命令行中运行):
- numpy版本不对
运行中的报错信息:
ModuleNotFoundError: No module named 'numpy.core._multiarray_umath'
ImportError: numpy.core.multiarray failed to import
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "<frozen importlib._bootstrap>", line 968, in _find_and_load
SystemError: <class '_frozen_importlib._ModuleLockManager'> returned a result with an error set
ImportError: numpy.core._multiarray_umath failed to import
ImportError: numpy.core.umath failed to import
2020-11-09 15:45:26.578761: F tensorflow/python/lib/core/bfloat16.cc:675] Check failed: PyBfloat16_Type.tp_base != nullptr
分析错误原因:
再利用conda install tensorflow-gpu安装时,安装好了一系列所依赖的库,此时运行正确。随后在pycharm中导入conda中配置好的虚拟环境,提示还缺少一些库如下图,进行了安装,注意此时安装后numpy库版本为1.15.4。
然后在再pycharm中的命令行模式运行代码后出现了上述错误。上面的错误是说numpy版本不对,需要更新。
改正方法:(更新的版本是原本装tensorflow库时自带的)
pip install --upgrade numpy==1.19.4
- cudnn版本不对
报错信息:
Traceback (most recent call last):
File "F:\anaconda\envs\tensorflow1\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "F:\anaconda\envs\tensorflow1\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "F:\anaconda\envs\tensorflow1\lib\site-packages\tensorflow\python\pywrap_tensorflo