问题:
1、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.
2、Failed to initialize NVML: Driver/library version mismatch
解决方法:reboot
Ubuntu GPU (Failed to initialize NVML: Driver/library version mismatch
最新推荐文章于 2024-07-29 15:54:22 发布