美文网首页
安装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