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