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   空明流光  阅读(115)  评论(0编辑  收藏  举报

相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 单线程的Redis速度为什么快?
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 展开说说关于C#中ORM框架的用法!

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5
点击右上角即可分享
微信分享提示