要在 Windows 上使用 CUDA 12.6 和 NVIDIA 驱动程序 560.94,您需要确保安装与之兼容的 PyTorch 版本。以下是找到适配版本的步骤:
-
访问 PyTorch 官方网站: 访问 PyTorch 官网。
-
选择合适的版本: 在首页上,您可以看到一个设置界面,您可以选择操作系统(Windows)、包管理工具(如 pip 或 conda)、语言(Python)和 CUDA 版本。在这里,选择支持 CUDA 12.6 的版本。
-
获取安装命令: 选择完毕后,页面下方会显示出适合您选择的 PyTorch 版本的安装命令。您可以复制这个命令来在命令行中执行。
-
安装 PyTorch: 打开命令提示符(CMD)或 Anaconda Prompt,粘贴并执行复制的命令。
如果在官方网站上找不到与 CUDA 12.6 兼容的 PyTorch 版本,您可能需要使用较旧的 CUDA 版本(例如 11.8)并确保更新驱动程序,以便与可用的 PyTorch 版本兼容。
如果有任何问题或安装过程中的错误,请提供具体的错误信息,以便我根据情况进一步帮助您!