Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available()

model = torch.load('mymodel.pt',map_location=torch.device('cpu'))
 
PREVIOUS NEXT
Tagged: #Attempting #deserialize #object #CUDA #device
ADD COMMENT
Topic
Name
4+9 =