
Adding visible gpu devices: 0 每次运行到这里卡很久
我们参考win中的想法.
参考1 :https://www.csdn.net/tags/MtTaMg1sODkzNjE2LWJsb2cO0O0O.html
在linux中,export CUDA_CACHE_MAXSIZE=4294967296 ,没想到真的变快了.
你可以通过如下命令测试是否有变快的感觉
import tensorflow as tf
a = tf.test.is_gpu_available()
print(a)
参考2: https://learn.foundry.com/nuke/content/comp_environment/air_tools/cc-intro.html#AIR
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)