轻松解决tensorflow GPU启动慢的问题

最近在用tf的时候,发现电脑的GPU启动非常慢,在Adding visible gpu devices: 0要卡很久,很慢。调用到了0号GPU,就是一只卡在那里.

要么就是下面一步卡顿很久

解决办法:

在环境变量->系统变量中添加CUDA_CACHE_MAXSIZE=6442450944,然后重启电脑。启动GPU就会变得飞快,可能重启后第一次会稍微缓慢一点。

注意后面这个值取决于不同电脑配置:显存*1024*1024*1024

比如说我的GPU是3060 6GB的显存 那么这个值就是6*1024*1024*1024

猜你喜欢

转载自blog.csdn.net/Sciws/article/details/125287168