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

轻松解决tensorflow GPU启动慢的问题_第1张图片

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

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

轻松解决tensorflow GPU启动慢的问题_第2张图片解决办法:

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

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

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

轻松解决tensorflow GPU启动慢的问题_第3张图片

你可能感兴趣的:(笔记,tensorflow,人工智能,python)