import os os.environ["CUDA_VISIBLE_DEVICES"] = "True" os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
可以翻译这句话:CUDA not available - defaulting to CPU. Note: This module is much faster with a GPU.
出现的是GPU版本和pytorch适用的问题。已经解决了
import os os.environ["CUDA_VISIBLE_DEVICES"] = "True" os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
可以翻译这句话:CUDA not available - defaulting to CPU. Note: This module is much faster with a GPU.
出现的是GPU版本和pytorch适用的问题。已经解决了