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())

 

posted on 2023-05-04 16:15  空明流光  阅读(106)  评论(0编辑  收藏  举报

导航