pytorch2.0.1对应的cuda
时间: 2023-09-21 13:08:48 浏览: 397
pytorch 查看cuda 版本方式
PyTorch 2.0.1 的 CUDA 版本取决于你的操作系统和 GPU 型号。你可以在 PyTorch 的官方网站上查找对应的 CUDA 版本。一般来说,PyTorch 2.0.1 支持 CUDA 9.2 和 CUDA 10.1 两个版本。如果你的 GPU 不支持这些 CUDA 版本,你可以考虑使用 CPU 版本的 PyTorch 进行开发。
阅读全文