摘要: (1)介绍 CUDA(Compute Unified Device Architecture) 是 NVIDIA 推出的一种并行计算平台和编程模型。它允许开发者利用 NVIDIA GPU(图形处理器)的强大计算能力进行通用计算,而不仅仅局限于图形渲染。eg:大量的矩阵运算等计算密集型任务 CUDA 阅读全文
posted @ 2024-11-29 15:55 卡卡发 阅读(33) 评论(0) 推荐(0) 编辑
摘要: 背景介绍: 有时候要复现其项目的实验时,环境需要尽量与原作者实验的环境靠近,这样结果浮现的也更加准确。 这里当需要安装torch==2.2.1+cu118等特定版本的时,出现了报错: 解决办法: 根据 PyTorch 官方推荐,可以使用以下命令来安装 torch 2.2.1+cu118 版本 pip 阅读全文
posted @ 2024-11-29 15:38 卡卡发 阅读(35) 评论(0) 推荐(0) 编辑