cuda12.1 对应pytorch和torchvision版本
时间: 2023-10-19 17:05:38 浏览: 172
对于 CUDA 12.1,PyTorch 1.7.0 及以上版本支持 CUDA 11.0、11.1 和 11.2,但不支持 CUDA 12.1。而 Torchvision 的版本则与 PyTorch 的版本相对应,所以也不支持 CUDA 12.1。如果需要使用 CUDA 12.1,建议等待 PyTorch 和 Torchvision 的更新版本。
相关问题
cuda11.7对应的pytorch和torchvision版本
目前还没有 cuda11.7 对应的官方 PyTorch 和 TorchVision 版本。但是,你可以使用 PyTorch 的 nightly build 版本,该版本具有 CUDA 11.4 和 11.5 的支持,因此可以在 CUDA 11.7 上运行。你可以通过以下命令安装 nightly build 版本:
```
pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html
```
需要注意的是,nightly build 版本是实验性版本,可能存在一些问题和 bug。如果你需要稳定的版本,请使用官方发布的 PyTorch 和 TorchVision 版本,并选择支持的 CUDA 版本。
cuda11.4 对应 pytorch、 torchvision、 torchaudio、 pytorch-cuda、 对应版本
以下是CUDA 11.4对应的PyTorch、TorchVision、Torchaudio和PyTorch CUDA的版本:
- PyTorch 1.9.0+cu111
- TorchVision 0.10.0+cu111
- Torchaudio 0.9.0+cu111
- PyTorch CUDA 11.1
请注意,PyTorch CUDA包是PyTorch的可选组件,需要单独安装。建议使用与PyTorch版本匹配的PyTorch CUDA版本。