安装TensorFlow-gpu的注意点

直接pip install tensorflow，安装到的是tensorflow-gpu 1.14.0，对应的CUDA版本是10.0，我为啥知道？本以为对应的是最新的CUDA10.1结果安好就悲剧了，还好软件提示：无法找到cudart64_100,请下载CUDA10.0，不过后面给的连接比较坑爹，给的是cuda9.0的，老实的我又中招了，下载了cuda9.0还是用不了。。。后来在下面找到了10.0。

GPU版本的对应关系：

补充较新的一组关系：

tensorflow-gpu 1.14.0 python3.5-3.7 ....... ......... 7.6.1 10.0

总结一下在已经安装好的win10下pycharm+python3.7下安装tensorflow的经历：

1、卸载CPU版本的tensorflow，途中遇到了无法卸载tensorboard和wrapt的情况，复制一下错误的提示，百度一下就解决了。

2、在CMD里面 pip install tensorflow-gpu还是比较顺利的安装好的

3、到pycharm里面settings->projectxxx里面看tensorflow-gpu的版本号。

4、随便写一个.py程序，import tensorflow as tf包含一下，然后下面会提示找不到cudart64_100,后面会给出提示和链接，进去下载对应版本的CUDA就可以，这里附上地址：https://developer.nvidia.com/cuda-toolkit-archive

5、下载cuDNN库，这个也得对应版本。下载地址：https://developer.nvidia.com/cudnn。解压后将

这三个复制合并到cuda目录下，win10默认路径为：

然后就可以用啦！

第一次使用会给出一堆提示信息，稍等一下就出结果。

019-06-29 20:04:09.909672: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1326] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 1391 MB memory) -> physical GPU (device: 0, name: GeForce 940MX, pci bus id: 0000:02:00.0, compute capability: 5.0)
相加: 0.300000

在笔记本上学习tensorflow基础，笔记本的显卡非常垃圾，不适合深度学习，这里仅作为说明，也算是体验一下过程。

参考链接：

https://blog.csdn.net/ifisher999/article/details/81085805

安装TensorFlow-gpu的注意点相关推荐

通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)
一. 安装环境 Ubuntu16.04.3 LST GPU: GeForce GTX1070 Python: 3.5 CUDA Toolkit 8.0 GA1 (Sept 2016) cuDNN v6 ...
linux安装tensorflow教程,Ubuntu 16.04 安装 TensorFlow(GPU支持)
本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow. 要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡. Tens ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
Docker 安装 TensorFlow GPU 实战
2019独角兽企业重金招聘Python工程师标准>>> Docker 安装 TensorFlow GPU 实战博客分类: 深度学习 docker 安装背景 AI如雨后春笋般的出现, ...
GTX860M安装Tensorflow gpu踩坑
GTX860M安装Tensorflow gpu踩坑前几天给台式成功配置了lightgbm的gpu版本和tensorflow的gpu版本,但是在单位还是要用LSTM,写一篇对应版本的踩坑文章日后查阅. ...
Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
ubuntu 16.04 安装TensorFlow GPU版本
推荐新版安装教程 http://blog.csdn.net/chenhaifeng2016/article/details/78874883 在ubuntu 16.04上安装cuda8.0和cudnn ...
Windows10安装TensorFlow GPU版本
本文描述的是如何在windows 10 64位操作系统安装TensorFlow,采用预编译版本. 前提条件一块支持CUDA 8.0的nvidia显卡. 可以在官方网站查询显卡的兼容性信息. 安装nv ...
windows安装tensorflow GPU
一.安装Anaconda Anaconda是Python发行包,包含了很多Python科学计算库.它是比直接安装Python更好的选择. 二.安装Tensorflow 如果安装了tensorflow, ...
CUDA安装 + tensorflow gpu版本出现的问题：Not creating XLA devices, tf_xla_enable_xla_devices not set
装完后用以下两行代码测试,结果竟然是False,输出了图片中的报错,肯定是哪装的有问题 import tensorflow as tf tf.test.is_gpu_available() 我的ten ...

安装TensorFlow-gpu的注意点

安装TensorFlow-gpu的注意点相关推荐

最新文章

热门文章