admin 管理员组

文章数量: 1086865

1060 5G/1065 版显卡安装TensorFlow/CUDA

前提

你的显卡是1060 5G版本，你想尝试一下TensorFlow，还好你遇到我。
我的系统win10 64位 ，python3.6，其他系统记得下载对应版本，我没有一一测试了。

准备工作

假设你已经安装好python3.6 ，且PIP过N个版本的TensorFlow，最后总会报一句：ImportError: Could not find ‘cudart64_90.dll’。
那么你可以接着看：

  1. 不要妄图去下载CUDA9.0了 ，因为1065的显卡只能支持CUDA9.2以上；
  2. 而NVIDIA 官方TensorFlow只支持CUDA9.0；
  3. 唯一的解决方案是安装非官方轮子TensorFlow1.10.0版本.10.0/py36/GPU/cuda92cudnn72avx2/tensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whl，我的是python3.6所以选择py36，如果你的是其他版本可以选择对应的py27等版本。
  4. 安装方法：pip3 install *.whl；
  5. 接着进 ，选择CUDA Toolkit 9.2，按照下图开始安装
  6. 再到后面的网址下载cuDNN7.2.1版本；
  7. 最后就是测试，一条条输入下列代码，没有报错就恭喜你吧，终于用1065尝到了TensorFlow！

import tensorflow as tf
hello = tf.constant(‘hello, tf!’)
sess = tf.Session()
print(sess.run(hello))
结果如下图：

本文标签： 1060 5G1065 版显卡安装TensorFlowCUDA

版权声明：本文标题：1060 5G1065 版显卡安装TensorFlowCUDA 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1687942684a160275.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。