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())
相关推荐
绒绒毛毛雨12 小时前
On the Plasticity and Stability for Post-Training Large Language Models
人工智能·机器学习·语言模型
ZTLJQ19 小时前
序列化的艺术:Python JSON处理完全解析
开发语言·python·json
H5css�海秀19 小时前
今天是自学大模型的第一天(sanjose)
后端·python·node.js·php
SuniaWang19 小时前
《Spring AI + 大模型全栈实战》学习手册系列 · 专题六:《Vue3 前端开发实战:打造企业级 RAG 问答界面》
java·前端·人工智能·spring boot·后端·spring·架构
阿贵---19 小时前
使用XGBoost赢得Kaggle比赛
jvm·数据库·python
无敌昊哥战神19 小时前
【LeetCode 257】二叉树的所有路径(回溯法/深度优先遍历)- Python/C/C++详细题解
c语言·c++·python·leetcode·深度优先
IDZSY043020 小时前
AI社交平台进阶指南:如何用AI社交提升工作学习效率
人工智能·学习
七七powerful20 小时前
运维养龙虾--AI 驱动的架构图革命:draw.io MCP 让运维画图效率提升 10 倍,使用codebuddy实战
运维·人工智能·draw.io
水星梦月20 小时前
大白话讲解AI/LLM核心概念
人工智能
温九味闻醉21 小时前
关于腾讯广告算法大赛2025项目分析1 - dataset.py
人工智能·算法·机器学习