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())
相关推荐
shenzhenNBA36 分钟前
python模块matplotlib绘图-饼图
python·matplotlib·pyplot·python绘制图表
buttonupAI7 小时前
今日Reddit各AI板块高价值讨论精选(2025-12-20)
人工智能
2501_904876488 小时前
2003-2021年上市公司人工智能的采纳程度测算数据(含原始数据+计算结果)
人工智能
咖啡の猫8 小时前
Python字典推导式
开发语言·python
曹文杰15190301128 小时前
2025 年大模型背景下应用统计本科 计算机方向 培养方案
python·线性代数·机器学习·学习方法
竣雄8 小时前
计算机视觉:原理、技术与未来展望
人工智能·计算机视觉
救救孩子把8 小时前
44-机器学习与大模型开发数学教程-4-6 大数定律与中心极限定理
人工智能·机器学习
Rabbit_QL8 小时前
【LLM评价指标】从概率到直觉:理解语言模型的困惑度
人工智能·语言模型·自然语言处理
Wulida0099918 小时前
建筑物表面缺陷检测与识别:基于YOLO11-C3k2-Strip模型的智能检测系统
python
呆萌很8 小时前
HSV颜色空间过滤
人工智能