pytorch、torchvision,cuda的对应关系

pytroch官方网站:https://pytorch.org/

  • 安装pytorch命令示例:

    pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
    
  • 2022年05月:windows RTX3080安装命令:
    电脑驱动版本为11.6,cuda=10.6 cudnn=10.x

pip install torch==1.10.1+cu113 torchvision=0.11.2+cu113 -f https://download.pytorch.org/whl/torch_stable.html

pytorch、torchvision,cu的对应关系

资料来源:https://codechina.csdn.net/mirrors/pytorch/vision?utm_source=csdn_github_accelerator
pytorch、torchvision,cuda的对应关系_第1张图片

可参考资源网站

https://download.pytorch.org/whl/torch_stable.html

pytorch、torchvision,cuda的对应关系_第2张图片

已成功安装:

  • 版本
  • torch 1.4.0+cu92
  • torchvision 0.5.0+cu92

你可能感兴趣的:(python开发,机器学习自学笔记,深度学习,pytorch)