参考链接
安装准备
注意事项
anaconda
安装路径不能出现汉字安装tensorflow
# GPU版本
pip3 install --upgrade tensorflow-gpu
# CPU版本
pip3 install --upgrade tensorflow
cuda
要找到对应的tensorflow版本,现在的tensorflow支持9.0,暂时不支持9.1,所以一定注意下载cuda 9.0
如果安装的是9.1版本,在import tensorflow as tf 的时候,报错:
import tensorflow as tf” ImportError: Could not find ‘cudart64_90.dll’测试一下是否安装成功,命令行输入 nvcc -V,看到版本信息就表示安装成功了
cuDNN
主要是添加环境变量,同时下载的时候注意与cuda版本的对应。解压到任意一个文件夹,**\cuda\bin 添加到环境变量中
如果已经安装了cuDNN,选择 升级cuDNN
error
1. 安装pip包thrift错误
Traceback (most recent call last):
File "d:\ProgramData\Anaconda3\lib\site-packages\pip\compat\__init__.py", line 73, in console_to_str
return s.decode(sys.__stdout__.encoding)
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xa1 in position 95: invalid start byte
解决方案:参考
# 修改73行
if sys.version_info >= (3,):
def console_to_str(s):
try:
#return s.decode(sys.__stdout__.encoding)
return s.decode('cp936') # 将上句代码改为此句代码
except UnicodeDecodeError:
return s.decode('utf_8')
2. h5py FutureWarning 问题修复
h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
from ._conv import register_converters as _register_converters
3. “import tensorflow as tf” ImportError: Could not find 'cudart64_90.dll'”
这个问题就是安装了cuda 9.1 而不是 9.0,只能 卸载 cuda 9.1, 重新安装cuda 9.0
Ubuntu 离线安装
TensorFlow 在Linux系统下离线安装
各种包的家
RedHat 7.2 离线安装tensorflow
两个教程足够
Ubuntu 在线安装
相关其他
- conda 使用清华大学开源软件镜像
- windows 查看GPU使用情况
只要在文件夹C:\Program Files\NVIDIA Corporation\NVSMI
里找到文件nvidia-smi.exe
,把该文件拖到命令提示符窗口(win+R,再输入‘CMD’进入),就可以显示关于GPU的信息,如下图所示: