NVIDIA CUDA12.0.133与pytorch2.0.1+cuda11.8匹配吗
时间: 2024-05-18 15:17:06 浏览: 296
torch_sparse-0.6.17+pt20cu118-cp38-cp38-win_amd64.whl.zip
根据官方文档,PyTorch 2.0.1 支持 CUDA 11.0、11.1、11.2、11.3、11.4、11.5、11.6 和 11.7 版本。因此,PyTorch 2.0.1 不支持 CUDA 12.0.133 版本。如果您想使用 PyTorch 2.0.1,建议您安装 CUDA 11.7 版本或更早的版本。
需要注意的是,CUDA 版本与 PyTorch 版本的兼容性是非常重要的。如果您安装了不兼容的 CUDA 版本,可能会导致 PyTorch 在 GPU 上无法正常工作,甚至可能会导致系统崩溃。因此,建议您仔细阅读官方文档,并确保CUDA版本与PyTorch版本兼容。
阅读全文