ubuntu系统,安装tensorflow-gpu 解决pip过慢、卡死情况

在这里假设您已经在ubuntu系统下安装好了NVIDAI、CUDA、cudnn等前提条件。
验证是否成功的代码:
检查驱动的命令 nvidia-smi
验证CUDA的命令 nvcc -V
查看cudnn的命令 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

之后可以通过下面一条命令可以快速安装tensorflow:
sudo python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.2.0

还需要注意一点,tf的版本需要匹配好cuda、cudnn的版本,下面是对应关系:

ubuntu系统,安装tensorflow-gpu 解决pip过慢、卡死情况_第1张图片

你可能感兴趣的:(ubuntu系统,安装tensorflow-gpu 解决pip过慢、卡死情况)