TensorFlow GPU 安装的问题处理
电脑配置
- win10 简体中文版
- gtx1060 显卡
过程
anaconda 安装
- 先安装anaconda集成了一些必要工具，比全部自己折腾好点。
不过还是踩到坑 【https://www.cnblogs.com/bikaqiou2000/p/11827881.html】 总结了踩坑的经历 - 完了后就安装好了 jupyter,conda 等必要工具
安装tensorflow cpu版
- 最初用pip安装tensorflow，各种问题，不行 不再细说
- 用conda 安装 cup版：如果只是安装cpu版 一般
conda install tensorflow
基本会成功安装最新版，不过用起来会慢很多
安装tensorflow gpu版
- 我选择安装gpu版
- 先看官方文档【https://tensorflow.google.cn/install/pip?lang=python3】，【https://tensorflow.google.cn/install/gpu】 里面说的必要条件要准备
- 开始安装，没有意外则可以成功
conda install tensorflow-gpu
- 安装要加速可以看看 这个【https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/】
- 然后运行下官方的代码
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
显然要翻车
安装nvidia相关工具
-
原因是NVIDIA 的相关工具还没有安装 接着看
-
下载并安装cuda 【https://developer.nvidia.com/cuda-toolkit】
-
下载cudnn 【https://developer.nvidia.com/rdp/cudnn-download】 中间官方网站要你注册和填写问卷，你就照做吧，不要钱的...
-
注意cuda toolkit 和 cudnn 版本一直性 10.1 cuda 就用 10.1 专属的 cudnn
-
下载cudnn 完了解压后将里面 三个文件夹 【bin，include，lib】放入cuda的安装目录。比如：【C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1】
-
然后再次运行下官方的代码
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
还是翻车，错误提示【CUDA driver version is insufficient for CUDA runtime version】
- 继续处理问题参考【https://www.jianshu.com/p/f85c30650028】 上面说我的显卡驱动版本太老了，更新前是 391.25，所以要更新最新驱动
C:Program FilesNVIDIA CorporationNVSMI> nvidia-smi
支持不到cuda 10
- 继续去官网 【https://www.nvidia.cn/Download/index.aspx?lang=cn#】 下载你的显卡对应驱动版本，我的gtx1060下载了441.2 可以支持到cuda 10.2
- 一路安装完最新驱动，再看看 【nvidia-smi】命令提示的显卡信息不是变化了
- 完了之后 执行 上面的tensorflow官方测试代码估计能成功了
...
tf.Tensor(-1986.1196, shape=(), dtype=float32)