UserWarning: Attempting to run cuBLAS, but there was no current CUDA context! Attempting to set the primary context...翻译
用户警告:PyTorch 准备调用 cuBLAS(CUDA 的线性代数库),当前线程还没有初始化 CUDA 上下文,PyTorch 正在帮你自动创建 CUDA 上下文。
人话
所以说意思就是:“你第一次在这个进程里真正用到 GPU,我现在帮你把 GPU 环境初始化一下。”
强迫症
如果你也和我一样强迫症发作,可以参考我的做法,在你定义DEVICE之后就加一句初始化torch.cuda.init()
DEVICE='cuda'iftorch.cuda.is_available()else'cpu'ifDEVICE=='cuda':torch.cuda.init()