PyTorch 2.0.1支持的CUDA版本?
时间: 2024-05-18 22:17:08 浏览: 442
pytorch3d全套安装流程
PyTorch 2.0.1支持的CUDA版本取决于您所使用的PyTorch版本和您的GPU驱动程序版本。您可以在PyTorch官方网站上查看每个PyTorch版本支持的CUDA版本。一般来说,PyTorch 2.0.1支持的CUDA版本为CUDA 10.1,但是在安装PyTorch之前,您需要确保已安装与您的GPU匹配的CUDA驱动程序和CUDA工具包。建议您在安装PyTorch之前查看官方文档以获得更详细的信息。
阅读全文