win10安装pytorch + cuda

1:下载cuda工具cuda-toolkit

地址:https://developer.nvidia.com/cuda-toolkit-archive

2:一路向下:安装

3:配置环境变量

path 环境变量中添加:

4:验证cuda是否安装成功:

nvcc -V

set cuda

5:安装cuDNN

登录cuDNN下载网页(必须注册账号并登录)并安装
cuDNN Archive | NVIDIA Developer

将解压后的文件夹中的bin、include、lib 三个文件夹,移动到CUDA Development 安装路径下,

与同名文件夹合并。

6:安装对应pytorch版本:

Start Locally | PyTorch

选择对应的版本,安装:

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

6:验证cuda是否成功:

python 复制代码
import torch
print(torch.cuda.is_available())
相关推荐
TF男孩9 小时前
重新认识Markdown:它不仅是排版工具,更是写Prompt的最佳结构
人工智能
想打游戏的程序猿9 小时前
AI时代的内容输出
人工智能
小兵张健10 小时前
Playwright MCP 截图标注方案调研:推荐方案 1
人工智能
凌杰11 小时前
AI 学习笔记:Agent 的能力体系
人工智能
IT_陈寒12 小时前
React状态管理终极对决:Redux vs Context API谁更胜一筹?
前端·人工智能·后端
舒一笑14 小时前
如何获取最新的技术趋势和热门技术
人工智能·程序员
聚客AI14 小时前
🎉OpenClaw深度解析:多智能体协同的三种模式、四大必装技能与自动化运维秘籍
人工智能·开源·agent
黄粱梦醒14 小时前
大模型企业级部署方案-vllm
人工智能·llm
IT_陈寒14 小时前
JavaScript代码效率提升50%?这5个优化技巧你必须知道!
前端·人工智能·后端
IT_陈寒14 小时前
Java开发必知的5个性能优化黑科技,提升50%效率不是梦!
前端·人工智能·后端