tensorflow2不支持1的api!!!!!
tensorflow2不支持1的api!!!!!
tensorflow2不支持1的api!!!!!
好了,首先看看自己的python版本,gpu的信息,tensorflow的版本,我下的最新python3.9直接报错*(没想到竟是噩梦的开始),py的版本应该在3.6-3.8
python 版本用 python -v看
gpu 信息用 nvidia-smi看
这里有个cuda version 11.2 不是你已经安装了11.2,而是你可以最高支持到11.2!
tensorflow的版本用 pip list看
根据你tensorflow下载对应的cuda和cudnn
下面是我2.4的对应标准
全部的表在https://tensorflow.google.cn/install/source_windows
图没截全,反正我大概就是
py:3.6-3.8
cudnn:8.0
cuda 11.0
这个样子
然后就开始准备安装!
一 cuda安装
首先从 控制面板所有控制面板项程序和功能 里把你能看到的和cuda有关的还有带frameview的nv软件卸载掉
下载cuda去官网下官网
重点!版本一定要一样!小数点也是!我下载的11.2根本不能用!还是要11.0才行!小数点也不能有差!这个巨tm关键!
但是nv官网你点下载toolkit就默认是11.1还是11.2来着了,我还没找历史的版本连接,没办法只能通过更改下载链接来实现下载历史版本的cuda了,比如网站的链接是
然后你就百度你要的包的版本加驱动号,我百度到我的是cuda_11.0.2_451.48_win10.exe就把下载链接里的对应数字改一改变成
下载完安装记住路径
添加环境变量
默认在c的话
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.0in
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.0extralib64
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.0include
打开cmd 输入nvcc -V
显示到你的cuda信息基本就是安装完了
二 cudnn安装
接着下载对应版本的cudnn,解压出来是一个cuda的文件夹,记住解压路径,比如D:cuda,把这个添加到环境变量
打开pycharm,下载tensorflow-gpu的包,慢的话自己去找镜像下载
试一试下面这个测试的代码,是知乎的一位老哥的 叫@爱生活的克劳德
import tensorflow as tf tf.compat.v1.disable_eager_execution() import os # 指定使用0,1,2三块卡 os.environ["CUDA_VISIBLE_DEVICES"] = "0" # Creates a graph. a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a') b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b') c = tf.matmul(a, b) # Creates a session with log_device_placement set to True. sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True)) # Runs the op. print(sess.run(c))
我改了一点,方便在ts2上跑
看到下面控制台你显示GPU0就说明你是GPU跑的,CPU0就是CPU跑的,如果是cpu跑的就再检查一下是不是版本不对,版本一定要按照表格来!小数点后面也要!
然后就成功了白字GPU0,