1. 进入 https://pytorch.org,依次选择 PyTorch Build -> Your OS ->Package ->Language ->Compute Platform,然后会生成安装命令或下载链接,执行或下载安装即可
如果没有 GPU,Compute Platform 选 CPU 即可
对于CUDA版本,可以执行cmd命令查看本地显卡支持的版本: nvidia-smi 或者打开 nvidia 相关的用户界面查看,下载的版本必须比这个版本低或等于这个版本
2. 安装 pytorch, 需要注意cuda版本和torch版本是对应的,详见以下链接,选择适合自己的
https://pytorch.org/get-started/previous-versions/
标签:Compute,版本,windows,Platform,pytorch,cuda,https From: https://www.cnblogs.com/nanfei/p/17371249.html