当前位置:首页|资讯

Windows 11 开发环境cuda12.6: 源码编译pytorch

作者:噼里啪啦哐当轰隆发布时间:2024-09-26

背景 我在具有 2 个 NVIDIA GPU 的 Windows 11 Pro 工作站上使用 VSCode 进行开发工作。我想在拥有裸机访问权限的同时使用 PyTorch 和 NVIDIA CUDA 工具包的最新功能。此外,我想使用新兴库进行编译,这些库为生成人工智能 (genai) 开发增加了功能并加快了速度。本文档概述了我为创建具有以下特征的环境所采取的步骤: Python 3.10NVIDIA CUDA 工具包 12.6NVIDIA cuDNN 9.4PyTorch 2.4Flash At...【查看原文】


Copyright © 2024 aigcdaily.cn  北京智识时代科技有限公司  版权所有  京ICP备2023006237号-1