windows 配置 cuda 和 pytorch
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/
3. 验证安装。执行以下 python 代码,如果输出为 True 说明配置成功
import torch print(torch.cuda.is_available())
桂棹兮兰桨,击空明兮溯流光。