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())
相关推荐
执笔论英雄7 分钟前
Slime异步原理(单例设计模式)4
开发语言·python·设计模式
愤怒的可乐14 分钟前
从零构建大模型智能体:统一消息格式,快速接入大语言模型
人工智能·语言模型·自然语言处理
每天一个java小知识2 小时前
AI Agent
人工智能
小徐敲java2 小时前
python使用s7协议与plc进行数据通讯(HslCommunication模拟)
开发语言·python
猫头虎2 小时前
如何解决 pip install 编译报错 fatal error: hdf5.h: No such file or directory(h5py)问题
人工智能·python·pycharm·开源·beautifulsoup·ai编程·pip
龙赤子2 小时前
人工智能AI的大框架
人工智能
比奥利奥还傲.2 小时前
本地+AI+大模型自由用!Cherry+Studio打破局域网限制
人工智能
雪碧聊技术2 小时前
深度学习、机器学习、人工智能三者的关系
人工智能·深度学习·机器学习
β添砖java2 小时前
机器学习初级
人工智能·机器学习
p***23362 小时前
python的sql解析库-sqlparse
数据库·python·sql