在Anaconda的base环境中,torch.cuda.is_available()函数是true,但是转为自己创建的虚拟环境后变为false;具体原因尚未找到,但是可以通过下面操作进行更正:
操作前:
torch.load(model_file)
操作后:
torch.load(model_file, map_location='cpu')
在Anaconda的base环境中,torch.cuda.is_available()函数是true,但是转为自己创建的虚拟环境后变为false;具体原因尚未找到,但是可以通过下面操作进行更正:
操作前:
torch.load(model_file)
操作后:
torch.load(model_file, map_location='cpu')