美文网首页
安装tensorflow-gpu 2.1遇到的问题与解决

安装tensorflow-gpu 2.1遇到的问题与解决

作者: 戴文KeepLearning | 来源:发表于2020-04-24 11:40 被阅读0次

    今天尝试用Anaconda安装tensorflow-gpu,版本2.1.0遇到了一些小问题,记录一下。

    1. pip安装的框架与conda环境不兼容

    pip install tensorflow-gpu
    安装好以后测试import tensorflow报错,好吧又只好删除掉。
    pip uninstall tensorflow-gpu
    ps. 貌似anaconda的虚拟环境里用pip安装深度学习框架总是会出现不兼容的情况。建议改用conda指令来安装。conda和pip安装包记得切换国内源,速度可以起飞。

    2. 将Anaconda某些包用迅雷下载到本地再利用conda安装

    conda install tensorflow-gpu
    会提示你安装一大堆的依赖包。这挺好的,能够保证兼容。
    然而某些包可能会出现下载时中断卡住的情况,然后下载安装不成功,比如我在安装依赖包cudnn-7.6.5-cuda10.1_0.conda的时候,170MB一直下载不下来,然后在命令里,我找到了清华源里该包的地址,拖到迅雷里下载,很快就下载下来了。

    微信图片_20200424113143.png

    这个时候,我们可以使用指令来安装本地的.conda包文件:

    conda install --use-local [file_path]
    [file_path]是.conda包文件的绝对路径。
    利用这种方式可以解决绝大多数包下载不稳定无法安装的问题!

    3. 因为h5pf报错

    测试用import tensorflow时,报错

    UserWarning: h5py is running against HDF5 1.10.5 when it was built against 1.10.4, this may cause problem
    

    这是conda里的h5pf包与安装的tensorflow-gpu 2.1.0不兼容引起的问题。
    (好像我被打脸了)
    解决办法是:利用pip删除掉现有的h5pf,然后重新安装一个...

    pip uninstall h5pf
    pip install h5pf
    

    4. CMD命令行输入Python报错

    图片.png

    在命令行进入Python时会出现以下报错,虽然不影响使用虚拟环境,但是看着不爽。
    解决方法:

    可以进入到该目录:[anaconda_virtual_env_path]\Lib\site-packages\pyreadline\lineeditor\history.py文件的第82行,修改如下:for line in open(filename, 'r',encoding='utf-8'):。在原有的open函数里新增了一个参数选项encoding='utf-8'就可以解决这个问题。

    相关文章

      网友评论

          本文标题:安装tensorflow-gpu 2.1遇到的问题与解决

          本文链接:https://www.haomeiwen.com/subject/wbixwhtx.html