需要安装 GPU 版本的 Torch 、 CUDA。
时间: 2024-05-17 11:16:30 浏览: 90
Windows+PyTorch GPU版本安装
5星 · 资源好评率100%
是的,如果您想在计算机的 GPU 上运行 Torch,您需要安装 GPU 版本的 Torch和CUDA。GPU版本的Torch是为了利用计算机的GPU进行加速计算,而CUDA则是GPU计算的核心库。在安装GPU版本的Torch之前,您需要确保您的计算机有NVIDIA GPU,并且已经安装了适当的NVIDIA驱动程序和CUDA工具包。您可以从Torch官方网站下载GPU版本的Torch和CUDA工具包,并按照安装指南进行安装。
阅读全文