企业官网建设流程全解析

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()