Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
raw
history blame contribute delete
198 Bytes
Add the following environment variable to the beginning of your code to switch to a CPU:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""
Another option is to get a better traceback from the GPU.