自用CUDA使用和xformers安装 文章目录 查看CUDA是否可用 快速使用CUDA 安装xformers 1.如果没有triton需要先安装triton 2.安装xformers 'tqdm_notebook' object has no attribute 'disp' 查看CUDA是否可用 查看GPU是否正确使用 # 检查CUDA是否可用 cuda_available = torch.cuda.is_available() print("CUDA available:", cuda_available) # 如果CUDA可用,则显示CUDA设备数量 if cuda_available: device_count = torch.cuda