原因是显存不足
1.选择不使用GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
2.降低batch_size.
3.缺少依赖
查看具体报错,具体问题单独解决。
此处我缺失了zlib123dllx64相关的库,下载后,写入path,重启后可以正常运行。
原因是显存不足
1.选择不使用GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
2.降低batch_size.
3.缺少依赖
查看具体报错,具体问题单独解决。
此处我缺失了zlib123dllx64相关的库,下载后,写入path,重启后可以正常运行。