cuda10.5对应的pytorch版本
时间: 2023-10-05 08:06:06 浏览: 83
pytorch-1.7.1-py3.8_cuda11.0.221_cudnn8.0.5_0.tar.bz2
CUDA 10.5 对应的 PyTorch 版本如下:
- PyTorch 1.8.0 和更高版本支持 CUDA 10.5
- PyTorch 1.7.0 支持 CUDA 10.5,但需要安装 cuTorch 1.7.0
建议使用最新的 PyTorch 版本,并根据自己的电脑配置选择对应的 CUDA 版本。
阅读全文