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

今天运行程序遇到一个问题，出现错误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=‘cpu’ to map your storages to the CPU.
然后百度查找解决方法，有很多的解决方案，比如出错语句为torch.load(model_file)，解决方案为model = torch.load(model_path, map_location='cpu')，然后试了一下发现在这并不适用，又出现了错误：RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /opt/conda/conda-bld/pytorch_1544174967633/work/aten/src/THC/THCGeneral.cpp:51
索性改回原来的版本，寻找其他解决方案，还有说是因为pytorch和cuda版本不一致，于是检查了一下，我安装的是pytorch1.0.0版本适合cuda版本是9.0，而环境中的cuda是10，但是一想更换cuda版本有点麻烦。所以再尝试一下其他解决办法，看到一个解决方案是

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'

突然唤起了我的回忆，打开自己的源程序，发现代码里边是

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '2'

而我现在的环境中用的是显卡1，所以我改成了

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

OK，问题解决。希望我的程序运行会有一个好结果呀！