pytorch==1.13.0 torchvision==0.14.0可以安装CUDA Toolkit 版本是 11.4吗
时间: 2023-10-02 22:13:12 浏览: 44
不可以。PyTorch 1.13.0 和 torchvision 0.14.0 不支持 CUDA Toolkit 11.4,您需要使用支持的 CUDA Toolkit 版本。根据 PyTorch 的官方文档,PyTorch 1.13.0 支持的 CUDA 版本为 10.2、10.1、9.2 和 9.0,而 torchvision 0.14.0 支持的 CUDA 版本为 10.2、10.1、9.2 和 9.0。
如果您想在 CUDA Toolkit 11.4 上运行 PyTorch 和 torchvision,则需要安装 PyTorch 1.9.0 或更高版本,以及 torchvision 0.10.0 或更高版本。这些版本已经支持 CUDA Toolkit 11.4。
相关问题
PyTorch 1.13.0 和 torchvision 0.14.0支持哪个cuda toolkit版本
PyTorch 1.13.0 和 torchvision 0.14.0 不支持 CUDA Toolkit 11.4,它们支持的 CUDA 版本列表如下:
- CUDA 10.2
- CUDA 10.1
- CUDA 9.2
- CUDA 9.0
如果您想使用 PyTorch 1.13.0 和 torchvision 0.14.0,您需要安装其中一个支持的 CUDA 版本。如果您已经安装了 CUDA Toolkit 11.4,则需要升级 PyTorch 和 torchvision 到支持 CUDA Toolkit 11.4 的版本。
torch==1.13.0+cu116,torchvision==0.14.0+cu116 ,torchaudio==0.13.0,需要安装cudatoolkit的哪个版本
根据您提供的PyTorch和TorchVision版本,推荐您安装CUDA 11.1或11.2版本,这两个版本的CUDA都支持PyTorch 1.13.0+cu116和TorchVision 0.14.0+cu116。另外,Torchaudio 0.13.0的官方文档中并未提供特定的CUDA版本要求,但是建议使用与PyTorch和TorchVision相同的CUDA版本。您可以在NVIDIA的官方网站上下载对应版本的CUDA和cudatoolkit。