cuda11.7对应的pytorch

PyTorch对应CUDA版本11.7是 PyTorch 1.8 以上版本, 需要在安装时指定对应CUDA版本, 如:

pip install torch torchvision torchaudio torchtext -f https://download.pytorch.org/whl/cu117/torch_stable.html

请注意您的系统及显卡驱动程序版本需要符合CUDA 11.7的要求。

相关问题

cuda 11.7对应pytorch版本

CUDA 11.7 目前还没有正式发布，最新的 CUDA 版本是 11.4，而 PyTorch 的支持情况如下：

- PyTorch 1.9.0 支持 CUDA 11.1、11.0、10.2、10.1、9.2 和 9.0
- PyTorch 1.8.0 支持 CUDA 11.1、11.0、10.2、10.1、9.2 和 9.0
- PyTorch 1.7.1 支持 CUDA 11.0、10.2、10.1、9.2 和 9.0

由于 CUDA 11.7 尚未发布，因此 PyTorch 目前还没有相应的支持版本。一般情况下，PyTorch 会在 CUDA 发布后的数周内发布相应的更新来支持新版本的 CUDA。

CUDA11.7对应pytorch版本与安装命令

CUDA 11.7 对应的 PyTorch 版本是 PyTorch 1.10.0。安装命令如下：

pip install torch==1.10.0+cu117 torchvision==0.11.1+cu117 torchaudio==0.10.0+cu117 -f https://download.pytorch.org/whl/cu117/torch_stable.html

请注意，此命令假设您已经安装了 CUDA 11.7 和相应的 CuDNN 版本。如果您尚未安装 CUDA 11.7 和 CuDNN，您需要先安装它们。