windows 配置 cuda pytorch

空明流光 / 2023-05-04 / 原文

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/