会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
公众号:NLP工程化
专注于Python/C++/CUDA、ML/DL/RL和NLP/KG/DS/LLM领域的技术分享。
博客园
首页
新随笔
联系
订阅
管理
2022年8月24日
PyTorch中的CUDA操作
摘要: CUDA(Compute Unified Device Architecture)是NVIDIA推出的异构计算平台,PyTorch中有专门的模块torch.cuda来设置和运行CUDA相关操作。本地安装环境为Windows10,Python3.7.8和CUDA 11.6,安装PyTorch最新稳定版
阅读全文
posted @ 2022-08-24 07:53 扫地升
阅读(2235)
评论(2)
推荐(1)