标签:environ 解决 when cpu device sha string dev 错误
原因:是因为gpu显存太小而导致的错误。
解决方法:
1.将batch_size改小
2.不用gpu训练,改用cpu训练
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
标签:environ 解决 when cpu device sha string dev 错误
原文地址:https://www.cnblogs.com/lpfeng/p/14535506.html