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适用的问题。已经解决了

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐