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())
相关推荐
shykevin26 分钟前
python开发Streamable HTTP MCP应用
开发语言·网络·python·网络协议·http
漫路在线1 小时前
JS逆向-某易云音乐下载器
开发语言·javascript·爬虫·python
Mr数据杨2 小时前
【Dv3Admin】插件 dv3admin_chatgpt 集成大语言模型智能模块
人工智能·语言模型·chatgpt
zm-v-159304339862 小时前
AI 赋能 Copula 建模:大语言模型驱动的相关性分析革新
人工智能·语言模型·自然语言处理
zhz52143 小时前
AI数字人融合VR全景:从技术突破到可信场景落地
人工智能·vr·ai编程·ai数字人·ai agent·智能体
数据与人工智能律师3 小时前
虚拟主播肖像权保护,数字时代的法律博弈
大数据·网络·人工智能·算法·区块链
成功人chen某4 小时前
配置VScodePython环境Python was not found;
开发语言·python
武科大许志伟4 小时前
武汉科技大学人工智能与演化计算实验室许志伟课题组参加2025中国膜计算论坛
人工智能·科技
哲讯智能科技4 小时前
【无标题】威灏光电&哲讯科技MES项目启动会圆满举行
人工智能
__Benco4 小时前
OpenHarmony平台驱动开发(十七),UART
人工智能·驱动开发·harmonyos