cuda11.1适配什么版本torch
时间: 2023-12-14 12:05:24 浏览: 18
cuda11.0 +对应版本cudnn
5星 · 资源好评率100%
如果你使用的是 CUDA 11.1,你可以使用 PyTorch 1.7.0 或更高版本。具体地,如果你想使用 CUDA 11.1 GPU 加速,你需要下载对应的 PyTorch 1.7.0+cu111 版本。如果你只使用 CPU,或者使用其他版本的 CUDA,可以下载相应的 PyTorch 版本。你可以在 PyTorch 的官方网站(https://pytorch.org/)上查找并下载适合你的 PyTorch 版本。同时,如果你使用的是 NVIDIA 显卡,你还需要安装对应版本的 NVIDIA CUDA Toolkit 和 cuDNN 库。
阅读全文