作者:手机用户2702935897 | 来源:互联网 | 2023-09-25 11:20
一分钟解决tensorflow2.0版本的GPU启动慢问题#最近在用tf的时候,发现电脑的GPU启动非常慢,调用到了0号GPU,就是一只卡在那里.如图从图中我们可以看到,我的GPU
一分钟解决tensorflow2.0版本的GPU启动慢问题
#最近在用tf的时候,发现电脑的GPU启动非常慢,调用到了0号GPU,就是一只卡在那里.
如图
从图中我们可以看到,我的GPU启动花了12分钟.
删掉安装的CUDA以后,非常快就跑出结果了,这是为什么呢?难道我们安装的不是GPU版本吗?
其实不是的,TF2.0已经把cpu和gpu合在一个包里面了,也就是说安装tf=2.1和tf-gpu=2.1是一样的.那么怎么解决呢?下面一分钟教你一分钟解决
首先打开windows的环境变量,然后添加一个变量,名字是CUDA_CACHE_MAXSIZE,数值是2147483648,再重启一下电脑就可以飞快的运行代码了.
如图
然后
然后把上面的名称和数值复制就好了.然后再重启一下电脑.