-
Notifications
You must be signed in to change notification settings - Fork 88
Closed
Description
在头一次初始化后,程序自动下载模型,但是在使用时缺提示没有CUDA 不能load模型。
所以像请问在哪可以直接下载CPU的模型 或者有什么设置吗
134 if not torch.cuda.is_available():
--> 135 raise RuntimeError('Attempting to deserialize object on a CUDA '
136 'device but torch.cuda.is_available() is False. '
137 'If you are running on a CPU-only machine, '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
Metadata
Metadata
Assignees
Labels
No labels